Skip to content

Commit 247cbcc

Browse files
Added test for asarray on sequence with arrays from difference devices
1 parent 750fc8b commit 247cbcc

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

dpctl/tests/test_tensor_asarray.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -321,3 +321,21 @@ def shape(self):
321321
x = dpt.asarray([d, d], sycl_queue=q)
322322
assert x.sycl_queue == q
323323
assert x.shape == (2,) + d.shape
324+
325+
326+
def test_asarray_seq_of_arrays_on_different_queues():
327+
q = get_queue_or_skip()
328+
329+
m = dpt.empty((2, 4), dtype="i2", sycl_queue=q)
330+
q2 = dpctl.SyclQueue()
331+
w = dpt.empty(4, dtype="i1", sycl_queue=q2)
332+
q3 = dpctl.SyclQueue()
333+
py_seq = [
334+
0,
335+
] * w.shape[0]
336+
res = dpt.asarray([m, [w, py_seq]], sycl_queue=q3)
337+
assert res.sycl_queue == q3
338+
assert dpt.isdtype(res.dtype, "integral")
339+
340+
with pytest.raises(dpctl.utils.ExecutionPlacementError):
341+
dpt.asarray([m, [w, py_seq]])

0 commit comments

Comments
 (0)