✨ HasDevice

Signed-off-by: nstarman <[email protected]>

Author: nstarman

src/array_api_typing/_array.py

@@ -67,9 +67,19 @@ def dtype(self, /) -> DTypeT_co:
         ...
 
 
+class HasDevice(Protocol):
+    """Protocol for array classes that have a device attribute."""
+
+    @property
+    def device(self) -> object:  # TODO: more specific type
+        """Hardware device the array data resides on."""
+        ...
+
+
 class Array(
     # ------ Attributes -------
     HasDType[DTypeT_co],
+    HasDevice,
     # ------- Methods ---------
     HasArrayNamespace[NamespaceT_co],
     # -------------------------