Skip to content

Commit 75a79e2

Browse files
committed
Add usm_type to test_buf in isin
1 parent c09b9ff commit 75a79e2

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

dpctl/tensor/_set_functions.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -744,7 +744,9 @@ def isin(
744744

745745
if test_dt != dt:
746746
# copy into C-contiguous memory, because the array will be flattened
747-
test_buf = dpt.empty_like(test_arr, dtype=dt, order="C")
747+
test_buf = dpt.empty_like(
748+
test_arr, dtype=dt, order="C", usm_type=res_usm_type
749+
)
748750
ht_ev, ev = _copy_usm_ndarray_into_usm_ndarray(
749751
src=test_arr, dst=test_buf, sycl_queue=exec_q, depends=dep_evs
750752
)

0 commit comments

Comments
 (0)