Added dtype compatibility check for device for result_type() and can_cast() functions.

Author: npolina4

dpctl/tensor/_manipulation_functions.py

@@ -436,7 +436,7 @@ def stack(arrays, axis=0):
     return res
 
 
-def can_cast(from_, to, casting="safe"):
+def can_cast(from_, to, casting="safe", device=None):
     """
     can_cast(from: usm_ndarray or dtype, to: dtype) -> bool
 
@@ -454,6 +454,23 @@ def can_cast(from_, to, casting="safe"):
 
     _supported_dtype([dtype_from, dtype_to])
 
+    if device is not None:
+        if isinstance(device, (dpctl.SyclQueue, dpt._device.Device)):
+            device = device.sycl_device
+        if not isinstance(device, dpctl.SyclDevice):
+            raise TypeError(f"Expected sycl_device type, got {type(device)}.")
+        if (
+            not device.has_aspect_fp16
+            and to == dpt.float16
+            or not device.has_aspect_fp64
+            and (to == dpt.float64 or to == dpt.complex128)
+        ):
+            return False
+        if not device.has_aspect_fp64 and (
+            to == dpt.complex64 or to == dpt.float32 and from_ is not complex
+        ):
+            return True
+
     return np.can_cast(dtype_from, dtype_to, casting)
 
 
@@ -475,6 +492,34 @@ def result_type(*arrays_and_dtypes):
     return np.result_type(*dtypes)
 
 
+def device_result_type(device, *arrays_and_dtypes):
+    """
+    device_result_type(device: sycl_device, arrays_and_dtypes: an arbitrary \
+        number usm_ndarrays or dtypes) -> dtype
+
+    Returns the dtype that results from applying the Type Promotion Rules to \
+        the arguments on current device.
+    """
+    dt = result_type(*arrays_and_dtypes)
+
+    if device is not None:
+        if isinstance(device, (dpctl.SyclQueue, dpt._device.Device)):
+            device = device.sycl_device
+        if not isinstance(device, dpctl.SyclDevice):
+            raise TypeError(f"Expected sycl_device type, got {type(device)}.")
+        if (
+            dt == dpt.float16
+            and not device.has_aspect_fp16
+            or dt == dpt.float64
+            and not device.has_aspect_fp64
+        ):
+            return dpt.float32
+        if dt == dpt.complex128 and not device.has_aspect_fp64:
+            return dpt.complex64
+
+    return dt
+
+
 def iinfo(dtype):
     """
     iinfo(dtype: integer data-type) -> iinfo_object