Add sycl queue validation for mask and values

Author: vlad-perevezentsev

dpnp/dpnp_iface_indexing.py

@@ -1870,8 +1870,17 @@ def putmask(a, /, mask, values):
         if a.dtype != values.dtype:
             values = dpnp.astype(values, a.dtype, casting="safe", copy=False)
 
-        _, exec_q = get_usm_allocations([a, mask, values])
-
+        exec_q = a.sycl_queue
+        if (
+            dpu.get_execution_queue(
+                [exec_q, mask.sycl_queue, values.sycl_queue]
+            )
+            is None
+        ):
+            raise ValueError(
+                "`mask` and `values` must be allocated on "
+                "the same SYCL queue as `a`"
+            )
         _manager = dpu.SequentialOrderManager[exec_q]
         dep_evs = _manager.submitted_events
 

dpnp/tests/test_indexing.py

@@ -1117,6 +1117,15 @@ def test_putmask_errors(self):
         # values as list
         assert_raises(TypeError, dpnp.putmask, a, mask, [1, 2, 3])
 
+        # values has a different SYCL queue
+        q1 = dpctl.SyclQueue()
+        q2 = dpctl.SyclQueue()
+        a = dpnp.arange(10, sycl_queue=q1)
+        mask = a > 3
+        val = dpnp.arange(5, sycl_queue=q2)
+        if q1 != q2:
+            assert_raises(ValueError, dpnp.putmask, a, mask, val)
+
 
 @pytest.mark.parametrize("m", [None, 0, 1, 2, 3, 4])
 @pytest.mark.parametrize("k", [-3, -2, -1, 0, 1, 2, 3])