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

now that needs rangeify 2 also

Browse Source
This commit is contained in:
George Hotz
2025-09-30 19:39:23 +08:00
parent ed5592b858
commit e7c7fdb47b
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
test/test_softmax_fusion.py
View File

@@ -44,6 +44,7 @@ class TestFuse(unittest.TestCase):
a = Tensor.rand(50,50).realize()
self._test_fuse(lambda a: a / a.mean(axis=1), a)
@unittest.skipIf(0<RANGEIFY<2, "needs RANGEIFY>1")
def test_fuse_argmax(self):
a = Tensor.rand(50,50).realize()
self._test_fuse(lambda a: a.argmax(axis=-1), a)