Set skip flag for linux job as well

More test disabling
pytorch · May 30, 2023 · 233e462 · 233e462
1 parent 94d49f3
commit 233e462
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 0 deletions.
diff --git a/.circleci/config.yml b/.circleci/config.yml
diff --git a/.circleci/config.yml.in b/.circleci/config.yml.in
@@ -355,6 +355,8 @@ jobs:
               TORCHAUDIO_TEST_ALLOW_SKIP_IF_CUDA_SMALL_MEMORY: true
               TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_AUDIO_OUT_DEVICE: true
               TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_MACOS: true
+              TORCHAUDIO_TEST_ALLOW_SKIP_IF_TEMPORARY_DISABLED: true
+
       - store_test_results:
           path: test-results
       - store_artifacts:

diff --git a/.github/workflows/unittest-linux-gpu.yml b/.github/workflows/unittest-linux-gpu.yml
@@ -74,6 +74,7 @@ jobs:
         export TORCHAUDIO_TEST_ALLOW_SKIP_IF_CUDA_SMALL_MEMORY=true
         export TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_HW_ACCEL=true
         export TORCHAUDIO_TEST_ALLOW_SKIP_IF_ON_PYTHON_310=true
+        export TORCHAUDIO_TEST_ALLOW_SKIP_IF_TEMPORARY_DISABLED=true
 
         declare -a args=(
             '-v'

diff --git a/test/torchaudio_unittest/prototype/conformer_wav2vec2_test.py b/test/torchaudio_unittest/prototype/conformer_wav2vec2_test.py
@@ -52,6 +52,8 @@ def test_pretrain_cpu_smoke_test(self, model, dtype):
         [torch.float32, torch.float64],
     )
     @skipIfNoCuda
+    # Disabled in CI: https://github.com/pytorch/audio/issues/3376
+    @disabledInCI
     def test_pretrain_cuda_smoke_test(self, model, dtype):
         model = model()
         self._smoke_test(model, torch.device("cuda"), dtype)