From e7c7fdb47b445a4b37a6f0c68025cd56f66a5128 Mon Sep 17 00:00:00 2001
From: George Hotz <geohot@gmail.com>
Date: Tue, 30 Sep 2025 19:39:23 +0800
Subject: [PATCH] now that needs rangeify 2 also

---
 test/test_softmax_fusion.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/test/test_softmax_fusion.py b/test/test_softmax_fusion.py
index 554f793d69..a86b3e40ee 100644
--- a/test/test_softmax_fusion.py
+++ b/test/test_softmax_fusion.py
@@ -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)