Skip to content

Commit 5637ff2

Browse files
Use dpctl.memory.as_usm_memory in __float__, etc.
1 parent e849d83 commit 5637ff2

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

dpctl/tensor/_usmarray.pyx

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -493,7 +493,8 @@ cdef class usm_ndarray:
493493

494494
def __bool__(self):
495495
if self.size == 1:
496-
return self.usm_data.copy_to_host().view(self.dtype).__bool__()
496+
mem_view = dpmem.as_usm_memory(self)
497+
return mem_view.copy_to_host().view(self.dtype).__bool__()
497498

498499
if self.size == 0:
499500
raise ValueError(
@@ -507,15 +508,17 @@ cdef class usm_ndarray:
507508

508509
def __float__(self):
509510
if self.size == 1:
510-
return self.usm_data.copy_to_host().view(self.dtype).__float__()
511+
mem_view = dpmem.as_usm_memory(self)
512+
return mem_view.copy_to_host().view(self.dtype).__float__()
511513

512514
raise ValueError(
513515
"only size-1 arrays can be converted to Python scalars"
514516
)
515517

516518
def __int__(self):
517519
if self.size == 1:
518-
return self.usm_data.copy_to_host().view(self.dtype).__int__()
520+
mem_view = dpmem.as_usm_memory(self)
521+
return mem_view.copy_to_host().view(self.dtype).__int__()
519522

520523
raise ValueError(
521524
"only size-1 arrays can be converted to Python scalars"

0 commit comments

Comments
 (0)