Added missing usm_type to tril() and triu() functions.

Author: npolina4

dpctl/tensor/_ctors.py

@@ -1247,19 +1247,31 @@ def tril(X, k=0):
 
     if k >= shape[nd - 1] - 1:
         res = dpt.empty(
-            X.shape, dtype=X.dtype, order=order, sycl_queue=X.sycl_queue
+            X.shape,
+            dtype=X.dtype,
+            order=order,
+            usm_type=X.usm_type,
+            sycl_queue=X.sycl_queue,
         )
         hev, _ = ti._copy_usm_ndarray_into_usm_ndarray(
             src=X, dst=res, sycl_queue=X.sycl_queue
         )
         hev.wait()
     elif k < -shape[nd - 2]:
         res = dpt.zeros(
-            X.shape, dtype=X.dtype, order=order, sycl_queue=X.sycl_queue
+            X.shape,
+            dtype=X.dtype,
+            order=order,
+            usm_type=X.usm_type,
+            sycl_queue=X.sycl_queue,
         )
     else:
         res = dpt.empty(
-            X.shape, dtype=X.dtype, order=order, sycl_queue=X.sycl_queue
+            X.shape,
+            dtype=X.dtype,
+            order=order,
+            usm_type=X.usm_type,
+            sycl_queue=X.sycl_queue,
         )
         hev, _ = ti._tril(src=X, dst=res, k=k, sycl_queue=X.sycl_queue)
         hev.wait()
@@ -1290,19 +1302,31 @@ def triu(X, k=0):
 
     if k > shape[nd - 1]:
         res = dpt.zeros(
-            X.shape, dtype=X.dtype, order=order, sycl_queue=X.sycl_queue
+            X.shape,
+            dtype=X.dtype,
+            order=order,
+            usm_type=X.usm_type,
+            sycl_queue=X.sycl_queue,
         )
     elif k <= -shape[nd - 2] + 1:
         res = dpt.empty(
-            X.shape, dtype=X.dtype, order=order, sycl_queue=X.sycl_queue
+            X.shape,
+            dtype=X.dtype,
+            order=order,
+            usm_type=X.usm_type,
+            sycl_queue=X.sycl_queue,
         )
         hev, _ = ti._copy_usm_ndarray_into_usm_ndarray(
             src=X, dst=res, sycl_queue=X.sycl_queue
         )
         hev.wait()
     else:
         res = dpt.empty(
-            X.shape, dtype=X.dtype, order=order, sycl_queue=X.sycl_queue
+            X.shape,
+            dtype=X.dtype,
+            order=order,
+            usm_type=X.usm_type,
+            sycl_queue=X.sycl_queue,
         )
         hev, _ = ti._triu(src=X, dst=res, k=k, sycl_queue=X.sycl_queue)
         hev.wait()