diff --git a/tinygrad/tensor.py b/tinygrad/tensor.py index c47d5df8e2..79154388e7 100644 --- a/tinygrad/tensor.py +++ b/tinygrad/tensor.py @@ -819,10 +819,10 @@ class Tensor: # in webgpu bool cannot be used as a storage buffer type def __lt__(self, x) -> Tensor: return mlops.Less.apply(*self._broadcasted(x, False)).cast(dtypes.float if self.device == "WEBGPU" else dtypes.bool) def __gt__(self, x) -> Tensor: return mlops.Less.apply(*self._broadcasted(x, True)).cast(dtypes.float if self.device == "WEBGPU" else dtypes.bool) - def __ge__(self, x) -> Tensor: return 1.0-(self Tensor: return 1.0-(self>x) + def __ge__(self, x) -> Tensor: return -(self Tensor: return -(self>x) def __ne__(self, x) -> Tensor: return (selfx) # type: ignore[override] - def __eq__(self, x) -> Tensor: return 1.0-(self != x) # type: ignore[override] + def __eq__(self, x) -> Tensor: return -(self != x) # type: ignore[override] # ***** functional nn ops *****