forked from tinygrad/tinygrad
putting add first orders loads as expected (#9991)
This commit is contained in:
@@ -301,7 +301,7 @@ class TestImageSimplification(unittest.TestCase):
|
||||
self.check(load,
|
||||
"((((idx2*2)+ridx0)<11)&((((idx1*8)+ridx1)<3)!=True))",
|
||||
"(((idx0+((idx1*512)+(ridx1*64)))+832)%1024)",
|
||||
"(((((idx1+((ridx1+5)//8))+1)//2)+((idx2*2)+ridx0))+-4)")
|
||||
"((((idx2*2)+ridx0)+(((idx1+((ridx1+5)//8))+1)//2))+-4)")
|
||||
|
||||
def test_simplify1(self):
|
||||
# idx has the form (A % m, A // m + k) and valid has (c0 < A) and (A < c1)
|
||||
|
||||
+1
-1
@@ -129,7 +129,7 @@ class Ops(FastEnum):
|
||||
WMMA = auto()
|
||||
|
||||
# BinaryOps
|
||||
MUL = auto(); SHL = auto(); SHR = auto(); IDIV = auto(); ADD = auto(); MAX = auto(); MOD = auto(); CMPLT = auto(); CMPNE = auto() # noqa: E702
|
||||
ADD = auto(); MUL = auto(); SHL = auto(); SHR = auto(); IDIV = auto(); MAX = auto(); MOD = auto(); CMPLT = auto(); CMPNE = auto() # noqa: E702
|
||||
XOR = auto(); OR = auto(); AND = auto(); THREEFRY = auto(); SUB = auto(); FDIV = auto(); POW = auto() # noqa: E702
|
||||
|
||||
# TernaryOps
|
||||
|
||||
Reference in New Issue
Block a user