github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

flip radd to be correct [run_process_replay] (#6390)

Browse Source
This commit is contained in:
George Hotz
2024-09-06 17:09:46 +08:00
committed by GitHub
parent 9a9fea7b8c
commit 6c15bd960b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/ops.py
View File

@@ -39,7 +39,7 @@ class MathTrait:
def ufix(self, x): return self.const_like(x) if not isinstance(x, MathTrait) else x
def __neg__(self): return self.ne(True) if getattr(self, 'dtype', None) == dtypes.bool else self*(-1)
def __add__(self, x): return self.alu(BinaryOps.ADD, self.ufix(x))
def __radd__(self, x): return self.alu(BinaryOps.ADD, self.ufix(x))
def __radd__(self, x): return self.ufix(x).alu(BinaryOps.ADD, self)
def __sub__(self, x): return self.alu(BinaryOps.ADD, self.ufix(-x))
def __rsub__(self, x): return self.ufix(x).alu(BinaryOps.ADD, -self)
def __mul__(self, x): return self.alu(BinaryOps.MUL, self.ufix(x))