update

Author: jcaip

test/prototype/test_sparse_api.py

@@ -31,7 +31,7 @@ class TestSemiStructuredSparse(common_utils.TestCase):
 
     @unittest.skipIf(not TORCH_VERSION_AT_LEAST_2_3, "pytorch 2.3+ feature")
     @unittest.skipIf(not torch.cuda.is_available(), "Need CUDA available")
-    @unittest.skipTest("Temporarily skipping to unpin nightlies")
+    @unittest.skip("Temporarily skipping to unpin nightlies")
     def test_sparse(self):
         input = torch.rand((128, 128)).half().cuda()
         model = (

test/sparsity/test_fast_sparse_training.py

@@ -31,7 +31,7 @@ class TestRuntimeSemiStructuredSparsity(TestCase):
     @unittest.skipIf(not TORCH_VERSION_AT_LEAST_2_4, "pytorch 2.4+ feature")
     @unittest.skipIf(not torch.cuda.is_available(), "Need CUDA available")
     @unittest.skipIf(is_fbcode(), "broken in fbcode")
-    @unittest.skipTest("Temporarily skipping to unpin nightlies")
+    @unittest.skip("Temporarily skipping to unpin nightlies")
     def test_runtime_weight_sparsification(self):
         # need this import inside to not break 2.2 tests
         from torch.sparse import SparseSemiStructuredTensorCUSPARSELT
@@ -73,7 +73,7 @@ def test_runtime_weight_sparsification(self):
     @unittest.skipIf(not TORCH_VERSION_AT_LEAST_2_4, "pytorch 2.4+ feature")
     @unittest.skipIf(not torch.cuda.is_available(), "Need CUDA available")
     @unittest.skipIf(is_fbcode(), "broken in fbcode")
-    @unittest.skipTest("Temporarily skipping to unpin nightlies")
+    @unittest.skip("Temporarily skipping to unpin nightlies")
     def test_runtime_weight_sparsification_compile(self):
         # need this import inside to not break 2.2 tests
         from torch.sparse import SparseSemiStructuredTensorCUSPARSELT