mawei/tinygrad
1
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-11 23:46:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

update test_where_removal (#15872)

Browse Source 
don't use UOp.ufix for const_like, it will broadcast dtype soon
This commit is contained in:
chenyu
2026-04-22 14:56:37 -04:00
committed by GitHub
parent 09ff3e1883
commit 3c8daa9a75
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/null/test_uop_symbolic.py
View File

@@ -853,7 +853,7 @@ class TestSymbolic(unittest.TestCase):
def test_where_removal(self):
cond = Variable("a", 0, 3) < 2
u1, u0 = cond.ufix(1), cond.ufix(0)
u1, u0 = cond.const_like(True), cond.const_like(False)
self.helper_test_variable(cond, 0, 1, "(a<2)")
self.helper_test_variable(cond.where(u1, u0), 0, 1, "(a<2)")
self.helper_test_variable(cond.where(u1, u0).where(u1, u0), 0, 1, "(a<2)")