Moves dpt.isdtype to _type_utils.py

Author: ndgrigorian

dpctl/tensor/__init__.py

@@ -51,7 +51,6 @@
     int16,
     int32,
     int64,
-    isdtype,
     uint8,
     uint16,
     uint32,
@@ -188,7 +187,7 @@
 )
 from ._sorting import argsort, sort
 from ._testing import allclose
-from ._type_utils import can_cast, finfo, iinfo, result_type
+from ._type_utils import can_cast, finfo, iinfo, isdtype, result_type
 
 __all__ = [
     "Device",

dpctl/tensor/_data_types.py

@@ -50,48 +50,6 @@
 complex128 = dtype("complex128")
 
 
-def isdtype(dtype_, kind):
-    """isdtype(dtype, kind)
-
-    Returns a boolean indicating whether a provided `dtype` is
-    of a specified data type `kind`.
-
-    See [array API](array_api) for more information.
-
-    [array_api]: https://data-apis.org/array-api/latest/
-    """
-
-    if not isinstance(dtype_, dtype):
-        raise TypeError(f"Expected instance of `dpt.dtype`, got {dtype_}")
-
-    if isinstance(kind, dtype):
-        return dtype_ == kind
-
-    elif isinstance(kind, str):
-        if kind == "bool":
-            return dtype_ == dtype("bool")
-        elif kind == "signed integer":
-            return dtype_.kind == "i"
-        elif kind == "unsigned integer":
-            return dtype_.kind == "u"
-        elif kind == "integral":
-            return dtype_.kind in "iu"
-        elif kind == "real floating":
-            return dtype_.kind == "f"
-        elif kind == "complex floating":
-            return dtype_.kind == "c"
-        elif kind == "numeric":
-            return dtype_.kind in "iufc"
-        else:
-            raise ValueError(f"Unrecognized data type kind: {kind}")
-
-    elif isinstance(kind, tuple):
-        return any(isdtype(dtype_, k) for k in kind)
-
-    else:
-        raise TypeError(f"Unsupported data type kind: {kind}")
-
-
 def _get_dtype(inp_dt, sycl_obj, ref_type=None):
     """
     Type inference utility to construct data type
@@ -121,7 +79,6 @@ def _get_dtype(inp_dt, sycl_obj, ref_type=None):
 __all__ = [
     "dtype",
     "_get_dtype",
-    "isdtype",
     "bool",
     "int8",
     "uint8",

dpctl/tensor/_type_utils.py

@@ -662,6 +662,48 @@ def _supported_dtype(dtypes):
     return True
 
 
+def isdtype(dtype, kind):
+    """isdtype(dtype, kind)
+
+    Returns a boolean indicating whether a provided `dtype` is
+    of a specified data type `kind`.
+
+    See [array API](array_api) for more information.
+
+    [array_api]: https://data-apis.org/array-api/latest/
+    """
+
+    if not isinstance(dtype, np.dtype):
+        raise TypeError(f"Expected instance of `dpt.dtype`, got {dtype}")
+
+    if isinstance(kind, dtype):
+        return dtype == kind
+
+    elif isinstance(kind, str):
+        if kind == "bool":
+            return dtype == np.dtype("bool")
+        elif kind == "signed integer":
+            return dtype.kind == "i"
+        elif kind == "unsigned integer":
+            return dtype.kind == "u"
+        elif kind == "integral":
+            return dtype.kind in "iu"
+        elif kind == "real floating":
+            return dtype.kind == "f"
+        elif kind == "complex floating":
+            return dtype.kind == "c"
+        elif kind == "numeric":
+            return dtype.kind in "iufc"
+        else:
+            raise ValueError(f"Unrecognized data type kind: {kind}")
+
+    elif isinstance(kind, tuple):
+        return any(isdtype(dtype, k) for k in kind)
+
+    else:
+        raise TypeError(f"Unsupported data type kind: {kind}")
+
+
 __all__ = [
     "_find_buf_dtype",
     "_find_buf_dtype2",
@@ -676,6 +718,7 @@ def _supported_dtype(dtypes):
     "can_cast",
     "finfo",
     "iinfo",
+    "isdtype",
     "result_type",
     "WeakBooleanType",
     "WeakIntegralType",