Merge pull request #1002 from IntelPython/fixed_dtype_full

Fixed error in cast dtype for full() function.

Author: oleksandr-pavlyk

dpctl/tensor/_ctors.py

@@ -774,14 +774,20 @@ def full(
 
     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))
+    fill_value_type = type(fill_value)
+    dtype = _get_dtype(dtype, sycl_queue, ref_type=fill_value_type)
     res = dpt.usm_ndarray(
         sh,
         dtype=dtype,
         buffer=usm_type,
         order=order,
         buffer_ctor_kwargs={"queue": sycl_queue},
     )
+    if fill_value_type in [float, complex] and np.issubdtype(dtype, np.integer):
+        fill_value = int(fill_value.real)
+    elif fill_value_type is complex and np.issubdtype(dtype, np.floating):
+        fill_value = fill_value.real
+
     hev, _ = ti._full_usm_ndarray(fill_value, res, sycl_queue)
     hev.wait()
     return res

dpctl/tests/test_usm_ndarray_ctor.py

@@ -1001,6 +1001,10 @@ def test_full_dtype_inference():
     assert np.issubdtype(dpt.full(10, 12.3).dtype, np.floating)
     assert np.issubdtype(dpt.full(10, 0.3 - 2j).dtype, np.complexfloating)
 
+    assert np.issubdtype(dpt.full(10, 12.3, dtype=int).dtype, np.integer)
+    assert np.issubdtype(dpt.full(10, 0.3 - 2j, dtype=int).dtype, np.integer)
+    assert np.issubdtype(dpt.full(10, 0.3 - 2j, dtype=float).dtype, np.floating)
+
 
 def test_full_fill_array():
     q = get_queue_or_skip()