We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3036a67 commit 2a90321Copy full SHA for 2a90321
dpctl/tensor/_ctors.py
@@ -363,6 +363,10 @@ def _copy_through_host_walker(seq_o, usm_res):
363
if isinstance(seq_o, dpt.usm_ndarray):
364
usm_res[...] = dpt.asnumpy(seq_o).copy()
365
return
366
+ if hasattr(seq_o, "__sycl_usm_array_interface__"):
367
+ usm_ar = _usm_ndarray_from_suai(seq_o)
368
+ usm_res[...] = dpt.asnumpy(usm_ar).copy()
369
+ return
370
if isinstance(seq_o, (list, tuple)):
371
for i, el in enumerate(seq_o):
372
_copy_through_host_walker(el, usm_res[i])
0 commit comments