fix version

pytorch · Sep 26, 2024 · a2519d8 · a2519d8
1 parent 46aeb9b
commit a2519d8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/sparsity/test_sparse_api.py b/test/sparsity/test_sparse_api.py
@@ -48,7 +48,7 @@ def test_sparse(self):
 
 class TestQuantSemiSparse(common_utils.TestCase):
 
-    @unittest.skipIf(not TORCH_VERSION_AT_LEAST_2_4, "pytorch 2.4+ feature")
+    @unittest.skipIf(not TORCH_VERSION_AT_LEAST_2_5, "pytorch 2.5+ feature")
     @unittest.skipIf(not torch.cuda.is_available(), "Need CUDA available")
     @common_utils.parametrize("compile", [True, False])
     def test_quant_semi_sparse(self, compile):