Added tests for full() function

Author: npolina4

dpctl/tensor/_ctors.py

@@ -762,18 +762,27 @@ def full(
     else:
         order = order[0].upper()
     dpctl.utils.validate_usm_type(usm_type, allow_none=True)
-    sycl_queue = normalize_queue_device(sycl_queue=sycl_queue, device=device)
 
     if isinstance(fill_value, (dpt.usm_ndarray, np.ndarray, tuple, list)):
+        if (
+            isinstance(fill_value, dpt.usm_ndarray)
+            and sycl_queue is None
+            and device is None
+        ):
+            sycl_queue = fill_value.sycl_queue
+        else:
+            sycl_queue = normalize_queue_device(
+                sycl_queue=sycl_queue, device=device
+            )
         X = dpt.asarray(
             fill_value,
             dtype=dtype,
-            device=device,
             usm_type=usm_type,
             sycl_queue=sycl_queue,
         )
         return dpt.broadcast_to(X, sh)
 
+    sycl_queue = normalize_queue_device(sycl_queue=sycl_queue, device=device)
     usm_type = usm_type if usm_type is not None else "device"
     dtype = _get_dtype(dtype, sycl_queue, ref_type=type(fill_value))
     res = dpt.usm_ndarray(

dpctl/tests/test_usm_ndarray_ctor.py

@@ -992,15 +992,36 @@ def test_full_dtype_inference():
 def test_full_fill_array():
     q = get_queue_or_skip()
 
-    Xnp = np.array([1, 2, 3], dtype=np.int32)
+    Xnp = np.array([1, 2, 3], dtype="i4")
     X = dpt.asarray(Xnp, sycl_queue=q)
 
     shape = (3, 3)
     Y = dpt.full(shape, X)
     Ynp = np.full(shape, Xnp)
 
+    assert Y.dtype == Ynp.dtype
+    assert Y.usm_type == "device"
     assert np.array_equal(dpt.asnumpy(Y), Ynp)
-    assert Ynp.dtype == Y.dtype
+
+
+def test_full_compute_follows_data():
+    q1 = get_queue_or_skip()
+    q2 = get_queue_or_skip()
+
+    X = dpt.arange(10, dtype="i4", sycl_queue=q1, usm_type="shared")
+    Y = dpt.full(10, X[3])
+
+    assert Y.dtype == X.dtype
+    assert Y.usm_type == X.usm_type
+    assert dpctl.utils.get_execution_queue((Y.sycl_queue, X.sycl_queue))
+    assert np.array_equal(dpt.asnumpy(Y), np.full(10, 3, dtype="i4"))
+
+    Y = dpt.full(10, X[3], dtype="f4", sycl_queue=q2, usm_type="host")
+
+    assert Y.dtype == dpt.dtype("f4")
+    assert Y.usm_type == "host"
+    assert dpctl.utils.get_execution_queue((Y.sycl_queue, q2))
+    assert np.array_equal(dpt.asnumpy(Y), np.full(10, 3, dtype="f4"))
 
 
 @pytest.mark.parametrize(