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

remove NEG from handwritten ast in tests (#6234)

Browse Source 
* remove NEG from handwritten ast in tests

* test_linearizer_failures
This commit is contained in:
chenyu
2024-08-22 09:06:59 -04:00
committed by GitHub
parent 1c5ef5b793
commit 3fc8203475
7 changed files with 75 additions and 107 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_linearizer.py
View File

@@ -592,6 +592,7 @@ class TestLinearizer(unittest.TestCase):
helper_linearizer_ast(sink, [dataset, idxs], wanna_output=[real_index])
# AssertionError: repeated stores in uops
@unittest.skip("TODO: rewrite without using NEG")
def test_argmax_multireduce_axis0(self):
t = Tensor.randn(10, 20).realize()
t_max = t.max((0,)).realize()
@@ -628,6 +629,7 @@ class TestLinearizer(unittest.TestCase):
UOp(UOps.SHAPETRACKER, None, arg=ShapeTracker(views=(View(shape=(1, 20, 1), strides=(0, 0, 0), offset=0, mask=None, contiguous=False),)), src=()),)),)),)),)) # noqa E501
helper_linearizer_ast(ast, [t, t_max], wanna_output=[real_argmax])
@unittest.skip("TODO: rewrite without using NEG")
def test_argmax_multireduce_flat(self):
t = Tensor.randn(10, 20).realize()
t_max = t.max().realize()