Added array support for fill_value for full() function.

Author: npolina4

dpctl/tensor/_ctors.py

@@ -726,7 +726,7 @@ def full(
     dtype=None,
     order="C",
     device=None,
-    usm_type="device",
+    usm_type=None,
     sycl_queue=None,
 ):
     """
@@ -761,10 +761,20 @@ def full(
         )
     else:
         order = order[0].upper()
-    dpctl.utils.validate_usm_type(usm_type, allow_none=False)
+    dpctl.utils.validate_usm_type(usm_type, allow_none=True)
     sycl_queue = normalize_queue_device(sycl_queue=sycl_queue, device=device)
-    if dtype is None and isinstance(fill_value, (dpt.usm_ndarray, np.ndarray)):
-        dtype = fill_value.dtype
+
+    if isinstance(fill_value, (dpt.usm_ndarray, np.ndarray, tuple, list)):
+        X = dpt.asarray(
+            fill_value,
+            dtype=dtype,
+            device=device,
+            usm_type=usm_type,
+            sycl_queue=sycl_queue,
+        )
+        return dpt.broadcast_to(X, sh)
+
+    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(
         sh,

dpctl/tests/test_usm_ndarray_ctor.py

@@ -992,12 +992,15 @@ def test_full_dtype_inference():
 def test_full_fill_array():
     q = get_queue_or_skip()
 
-    dtype = np.int32
-    X = dpt.full(10, dpt.usm_ndarray(1, dtype=dtype), sycl_queue=q)
-    assert dtype == X.dtype
+    Xnp = np.array([1, 2, 3], dtype=np.int32)
+    X = dpt.asarray(Xnp, sycl_queue=q)
 
-    X = dpt.full(10, np.ndarray(1, dtype=dtype), sycl_queue=q)
-    assert dtype == X.dtype
+    shape = (3, 3)
+    Y = dpt.full(shape, X)
+    Ynp = np.full(shape, Xnp)
+
+    assert np.array_equal(dpt.asnumpy(Y), Ynp)
+    assert Ynp.dtype == Y.dtype
 
 
 @pytest.mark.parametrize(