tc

test/inductor/test_cuda_select_algorithm.py

@@ -17,7 +17,11 @@ from torch.testing._internal.common_device_type import (
 from torch.testing._internal.common_quantized import (
     _calculate_dynamic_per_channel_qparams,
 )
-from torch.testing._internal.common_utils import parametrize, TEST_CUDA
+from torch.testing._internal.common_utils import (
+    parametrize,
+    TEST_CUDA,
+    TEST_WITH_SLOW_GRADCHECK,
+)
 
 
 try:
@@ -79,6 +83,7 @@ class TestSelectAlgorithmCuda(BaseTestSelectAlgorithm):
     @parametrize("in_features", (128, 144, 1024))
     @parametrize("out_features", (64, 65, 1024))
     @unittest.skipIf(not TEST_CUDA, "CUDA not available")
+    @unittest.skipIf(TEST_WITH_SLOW_GRADCHECK, "Leaking memory")
     def test_int8_woq_mm_cuda(
         self, dtype, batch_size, mid_dim, in_features, out_features
     ):