Skip to content

Commit 7f14dfc

Browse files
Update _dlpack.pyx to use dpctl_ext.tensor
1 parent 18c3d61 commit 7f14dfc

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

dpctl_ext/tensor/_dlpack.pyx

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1209,7 +1209,9 @@ def from_dlpack(x, /, *, device=None, copy=None):
12091209
)
12101210
return from_dlpack_capsule(cpu_caps)
12111211
else:
1212-
import dpctl.tensor as dpt
1212+
# TODO: revert to `import dpctl.tensor`
1213+
# when dpnp fully migrates dpctl/tensor
1214+
import dpctl_ext.tensor as dpt
12131215
return dpt.asarray(blob, device=dev)
12141216
elif got_buffer_error:
12151217
# we are here, because dlpack_attr could not deal with requested

0 commit comments

Comments
 (0)