Define __hash__

Author: betatim

array_api_strict/_array_object.py

@@ -53,8 +53,14 @@ def __repr__(self):
         return f"array_api_strict.Device('{self._device}')"
 
     def __eq__(self, other):
+        if not isinstance(other, Device):
+            return False
         return self._device == other._device
 
+    def __hash__(self):
+        return hash(("Device", self._device))
+
+
 CPU_DEVICE = Device()
 
 _default = object()

array_api_strict/_creation_functions.py

@@ -307,8 +307,11 @@ def meshgrid(*arrays: Array, indexing: str = "xy") -> List[Array]:
     if len({a.dtype for a in arrays}) > 1:
         raise ValueError("meshgrid inputs must all have the same dtype")
 
+    if len({a.device for a in arrays}) > 1:
+        raise ValueError("meshgrid inputs must all be on the same device")
+
     return [
-        Array._new(array, device=device)
+        Array._new(array, device=array.device)
         for array in np.meshgrid(*[a._array for a in arrays], indexing=indexing)
     ]