Skip to content

Commit d8fa3b2

Browse files
Test from_dlpack on numpy input with strides not multiple of elementsize
1 parent 6f4cd55 commit d8fa3b2

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

dpctl/tests/test_usm_ndarray_dlpack.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -719,3 +719,15 @@ def test_copy_via_host():
719719
z = dpt.from_dlpack(x, device=(x_dl_dev[0], j))
720720
assert isinstance(z, dpt.usm_ndarray)
721721
assert z.usm_type == "device"
722+
723+
724+
def test_copy_via_host_gh_1789():
725+
"Test based on review example from gh-1789"
726+
get_queue_or_skip()
727+
x_np = np.ones((10, 10), dtype="i4")
728+
# strides are no longer multiple of itemsize
729+
x_np.strides = (x_np.strides[0] - 1, x_np.strides[1])
730+
with pytest.raises(BufferError):
731+
dpt.from_dlpack(x_np)
732+
with pytest.raises(BufferError):
733+
dpt.from_dlpack(x_np, device=(14, 0))

0 commit comments

Comments
 (0)