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

Revert "add UOps.VALID (#6387)" (#6441)

Browse Source 
This reverts commit 8186e4e7d6.
This commit is contained in:
George Hotz
2024-09-09 21:33:00 +08:00
committed by GitHub
parent e7dd08448f
commit dbd4536167
9 changed files with 20 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/test_multitensor.py
View File

@@ -572,13 +572,13 @@ class TestMultiTensor(unittest.TestCase):
assert ast.op is UOps.STORE
assert ast.src[2].arg is BinaryOps.ADD
assert ast.src[2].src[0].op is UOps.LOAD and ast.src[2].src[0]
assert ast.src[2].src[1].src[1].op is UOps.CONST and ast.src[2].src[1].src[1].arg == 1
assert ast.src[2].src[1].op is UOps.CONST and ast.src[2].src[1].arg == 1
t = 2 * t
for si in t.schedule():
ast = si.ast.src[0]
assert ast.op is UOps.STORE
assert ast.src[2].arg is BinaryOps.MUL
assert ast.src[2].src[0].src[1].op is UOps.CONST and ast.src[2].src[0].src[1].arg == 2
assert ast.src[2].src[0].op is UOps.CONST and ast.src[2].src[0].arg == 2
assert ast.src[2].src[1].op is UOps.LOAD
t = t + t.full_like(3)
for si in t.schedule():
@@ -586,7 +586,7 @@ class TestMultiTensor(unittest.TestCase):
assert ast.op is UOps.STORE
assert ast.src[2].arg is BinaryOps.ADD
assert ast.src[2].src[0].op is UOps.LOAD
assert ast.src[2].src[1].src[1].op is UOps.CONST and ast.src[2].src[1].src[1].arg == 3
assert ast.src[2].src[1].op is UOps.CONST and ast.src[2].src[1].arg == 3
def test_shard_memory(self):
devices = (d0, d1, d2, d3)