github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 15:38:29 -05:00
Code Issues Packages Projects Releases Wiki Activity

add rlshift and rrshift special methods (#7185)

Browse Source
This commit is contained in:
ignaciosica
2024-10-21 09:37:02 -03:00
committed by GitHub
parent 8f375b71c5
commit 5551cf6689
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/ops.py
View File

@@ -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))