Skip to content

Commit f8a2e3e

Browse files
Change exception raised by __dlpack__ if dl_device is unsupported
It used to raise NotImplementedError, not raises BufferError
1 parent 39931b2 commit f8a2e3e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

dpctl/tensor/_usmarray.pyx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1242,7 +1242,7 @@ cdef class usm_ndarray:
12421242
_arr.flags["W"] = self.flags["W"]
12431243
return c_dlpack.numpy_to_dlpack_versioned_capsule(_arr, True)
12441244
else:
1245-
raise NotImplementedError(
1245+
raise BufferError(
12461246
f"targeting `dl_device` {dl_device} with `__dlpack__` is not "
12471247
"yet implemented"
12481248
)

0 commit comments

Comments
 (0)