clean

Author: ev-br

array_api_strict/_devices.py

@@ -30,6 +30,7 @@ def __hash__(self) -> int:
         return hash(("Device", self._device))
 
     def _supported_dtypes(self) -> list[DType]:
+        # XXX useful? Unused ATM
         return list(dt for dt in _all_dtypes if device_supports_dtype(self, dt))
 
 
@@ -47,6 +48,9 @@ def check_device(device: Device | None) -> None:
         raise ValueError(f"Unsupported device {device!r}")
 
 
+# Helpers for device-specific dtype support
+
+
 def device_supports_dtype(device: Device | None, dtype: DType |None) -> bool:
     """True if `device` supports `dtype`, False otherwise."""
     # special-case F32_device
@@ -57,9 +61,7 @@ def device_supports_dtype(device: Device | None, dtype: DType |None) -> bool:
     return True
 
 
-# Device-specific dtype maps
-
-def _map_helper(dtypes: list[DType], device: Device) -> dict[str, DType]:
+def _map_supported(dtypes: list[DType], device: Device) -> dict[str, DType]:
     return {
         dt._canonic_name: dt
         for dt in dtypes
@@ -81,29 +83,3 @@ def _map_helper(dtypes: list[DType], device: Device) -> dict[str, DType]:
     "numeric": _numeric_dtypes
 }
 
-_device_dtype_maps = {}
-
-for device in ALL_DEVICES:
-    _map = {}
-    for kind, dtypes in _kind_to_dtypes.items():
-        _map[kind] = {
-            dt._canonic_name: dt
-            for dt in dtypes
-            if device_supports_dtype(device, dt)
-        }
-    _device_dtype_maps[device] = _map
-
-
-
-'''
-    _kind_to_dtypes = {
-        None: {x._canonic_name:x for x in _all_dtypes},
-        "bool": {"bool": bool},
-        "signed integer": {x._canonic_name: x for x in _signed_integer_dtypes},
-        "unsigned integer": {x._canonic_name: x for x in _unsigned_integer_dtypes},
-        "integral": {x._canonic_name: x for x in _integer_dtypes},
-        "real floating": {x._canonic_name: x for x in _real_floating_dtypes},
-        "complex floating": {x._canonic_name: x for x in _complex_floating_dtypes},
-        "numeric": {x._canonic_name: x for x in _numeric_dtypes}
-    }
-'''

array_api_strict/_info.py

@@ -63,7 +63,7 @@ def dtypes(
                 dtypes = _devices._kind_to_dtypes[kind]
             except KeyError:
                 raise ValueError(f"unsupported kind: {kind!r}")  
-            res = _devices._map_helper(dtypes, device)
+            res = _devices._map_supported(dtypes, device)
             return res
 
         elif isinstance(kind, tuple):