exactly specify excluded distributed/pipeline/sync tests

jfgrimm · Feb 2, 2024 · bf06a6c · bf06a6c
1 parent 16b2044
commit bf06a6c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/easybuild/easyconfigs/p/PyTorch/PyTorch-2.1.2-foss-2023a-CUDA-12.1.1.eb b/easybuild/easyconfigs/p/PyTorch/PyTorch-2.1.2-foss-2023a-CUDA-12.1.1.eb
@@ -178,7 +178,12 @@ excluded_tests = {
         'distributed/rpc/test_tensorpipe_agent',
         # failures on OmniPath systems, which don't support some optional InfiniBand features
         # See https://github.com/pytorch/tensorpipe/issues/413
-        'distributed/pipeline/sync',
+        'distributed/pipeline/sync/skip/test_gpipe',
+        'distributed/pipeline/sync/skip/test_leak',
+        'distributed/pipeline/sync/test_bugs',
+        'distributed/pipeline/sync/test_inplace',
+        'distributed/pipeline/sync/test_pipe',
+        'distributed/pipeline/sync/test_transparency',
     ]
 }