Skip distributed tests if not supported (pytorch#2004)

Joyce94 · Jul 7, 2017 · ebdec9a · ebdec9a
1 parent c3c7845
commit ebdec9a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/test/test_distributed.py b/test/test_distributed.py
@@ -17,6 +17,11 @@
 MASTER_ADDR = '127.0.0.1'
 
 
+if not dist.is_available():
+    print('Distributed not available, skipping tests')
+    sys.exit(0)
+
+
 @contextmanager
 def _lock():
     lockfile = os.path.join(TEMP_DIR, 'lockfile')