diff --git a/tinygrad/ops.py b/tinygrad/ops.py index 69fe7ec3ce..abcb8b64e5 100644 --- a/tinygrad/ops.py +++ b/tinygrad/ops.py @@ -55,7 +55,9 @@ class MathTrait: def __mul__(self, x): return self.alu(BinaryOps.MUL, self.ufix(x)) def __rmul__(self, x): return self.ufix(x).alu(BinaryOps.MUL, self) def __lshift__(self, x): return self.alu(BinaryOps.SHL, self.ufix(x)) + def __rlshift__(self, x): return self.ufix(x).alu(BinaryOps.SHL, self) def __rshift__(self, x): return self.alu(BinaryOps.SHR, self.ufix(x)) + def __rrshift__(self, x): return self.ufix(x).alu(BinaryOps.SHR, self) def __floordiv__(self, x): return self.alu(BinaryOps.IDIV, self.ufix(x)) def __rfloordiv__(self, x): return self.ufix(x).alu(BinaryOps.IDIV, self) def __truediv__(self, x): return self.alu(BinaryOps.MUL, self.ufix(x).alu(UnaryOps.RECIP))