File tree Expand file tree Collapse file tree 2 files changed +8
-3
lines changed
infrastructure/frameworks Expand file tree Collapse file tree 2 files changed +8
-3
lines changed Original file line number Diff line number Diff line change @@ -20,7 +20,6 @@ input_args = [
2020array_args = [
2121 " data" ,
2222]
23- expected_failure_implementations = [" sycl" ]
2423
2524[benchmark .parameters .S ]
2625n_samples = 1024
Original file line number Diff line number Diff line change @@ -76,9 +76,15 @@ def copy_from_func(self) -> Callable:
7676 any array created by the framework possibly on
7777 a device memory domain."""
7878
79- import dpctl .tensor as dpt
79+ def _copy_from_func (usm_array ):
80+ import dpctl .tensor as dpt
81+
82+ if isinstance (usm_array , dpt .usm_ndarray ):
83+ return dpt .asnumpy (usm_array )
84+ else :
85+ return usm_array
8086
81- return dpt . asnumpy
87+ return _copy_from_func
8288
8389 def version (self ) -> str :
8490 """Returns the framework version."""
You can’t perform that action at this time.
0 commit comments