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

Author: npolina4

dpctl/tensor/__init__.py

@@ -48,6 +48,7 @@
     broadcast_to,
     can_cast,
     concat,
+    device_result_type,
     expand_dims,
     finfo,
     flip,
@@ -137,4 +138,5 @@
     "get_print_options",
     "set_print_options",
     "print_options",
+    "device_result_type",
 ]

dpctl/tensor/_manipulation_functions.py

@@ -455,7 +455,7 @@ def can_cast(from_, to, casting="safe", device=None):
     _supported_dtype([dtype_from, dtype_to])
 
     if device is not None:
-        if isinstance(device, (dpctl.SyclQueue, dpt._device.Device)):
+        if isinstance(device, (dpctl.SyclQueue, dpt.Device)):
             device = device.sycl_device
         if not isinstance(device, dpctl.SyclDevice):
             raise TypeError(f"Expected sycl_device type, got {type(device)}.")
@@ -503,7 +503,7 @@ def device_result_type(device, *arrays_and_dtypes):
     dt = result_type(*arrays_and_dtypes)
 
     if device is not None:
-        if isinstance(device, (dpctl.SyclQueue, dpt._device.Device)):
+        if isinstance(device, (dpctl.SyclQueue, dpt.Device)):
             device = device.sycl_device
         if not isinstance(device, dpctl.SyclDevice):
             raise TypeError(f"Expected sycl_device type, got {type(device)}.")