File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff line change @@ -493,7 +493,8 @@ cdef class usm_ndarray:
493
493
494
494
def __bool__ (self ):
495
495
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__()
497
498
498
499
if self .size == 0 :
499
500
raise ValueError (
@@ -507,15 +508,17 @@ cdef class usm_ndarray:
507
508
508
509
def __float__ (self ):
509
510
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__()
511
513
512
514
raise ValueError (
513
515
" only size-1 arrays can be converted to Python scalars"
514
516
)
515
517
516
518
def __int__ (self ):
517
519
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__()
519
522
520
523
raise ValueError (
521
524
" only size-1 arrays can be converted to Python scalars"
You can’t perform that action at this time.
0 commit comments