Merge pull request #2450 from Flamefire/pytorch_skip_bottleneck

Exclude bottleneck tests in PyTorch
easybuilders · Jun 1, 2021 · 835a5c6 · 835a5c6
2 parents 19a8427 + 1a4e76b
commit 835a5c6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/easybuild/easyblocks/p/pytorch.py b/easybuild/easyblocks/p/pytorch.py
@@ -225,6 +225,8 @@ def test_step(self):
         env.setvar('XDG_CACHE_HOME', os.path.join(self.tmpdir, '.cache'))
         # Pretend to be on FB CI which disables some tests, especially those which download stuff
         env.setvar('SANDCASTLE', '1')
+        # Skip this test(s) which is very flaky
+        env.setvar('SKIP_TEST_BOTTLENECK', '1')
         # Parse excluded_tests and flatten into space separated string
         excluded_tests = []
         for arch, tests in self.cfg['excluded_tests'].items():