Merge pull request #1051 from IntelPython/change_linspace_int

Changed generating integer output for linspace() function.

Author: oleksandr-pavlyk

dpctl/tensor/_ctors.py

@@ -1119,25 +1119,28 @@ def linspace(
     num = operator.index(num)
     if num < 0:
         raise ValueError("Number of points must be non-negative")
-    ((start, stop,), dt) = _coerce_and_infer_dt(
+    _, dt = _coerce_and_infer_dt(
         start,
         stop,
         dt=dtype,
         sycl_queue=sycl_queue,
         err_msg="start and stop must be Python scalars.",
         allow_bool=True,
     )
-    if dtype is None and np.issubdtype(dt, np.integer):
+    int_dt = None
+    if np.issubdtype(dt, np.integer):
+        if dtype is not None:
+            int_dt = dt
         dt = ti.default_device_fp_type(sycl_queue)
         dt = dpt.dtype(dt)
         start = float(start)
         stop = float(stop)
-    res = dpt.empty(num, dtype=dt, sycl_queue=sycl_queue)
+    res = dpt.empty(num, dtype=dt, usm_type=usm_type, sycl_queue=sycl_queue)
     hev, _ = ti._linspace_affine(
         start, stop, dst=res, include_endpoint=endpoint, sycl_queue=sycl_queue
     )
     hev.wait()
-    return res
+    return res if int_dt is None else dpt.astype(res, int_dt)
 
 
 def eye(

dpctl/tests/test_usm_ndarray_ctor.py

@@ -1316,6 +1316,13 @@ def test_linspace_fp_max(dtype):
     )
 
 
+def test_linspace_int():
+    q = get_queue_or_skip()
+    X = dpt.linspace(0.1, 9.1, 11, endpoint=True, dtype=int, sycl_queue=q)
+    Xnp = np.linspace(0.1, 9.1, 11, endpoint=True, dtype=int)
+    assert np.array_equal(dpt.asnumpy(X), Xnp)
+
+
 @pytest.mark.parametrize(
     "dt",
     _all_dtypes,