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(01") def test_fuse_argmax(self): a = Tensor.rand(50,50).realize() self._test_fuse(lambda a: a.argmax(axis=-1), a)