Enabled gloo backend in test_distributed unit tests for ROCm (#40395)

Summary: Pull Request resolved: #40395 Reviewed By: ngimel Differential Revision: D25181692 Pulled By: mrshenli fbshipit-source-id: 29f478c974791efc0acea210c8c9e574944746a5
pytorch · Nov 26, 2020 · f1c9856 · f1c9856
1 parent db1b0b0
commit f1c9856
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/run_test.py b/test/run_test.py
@@ -263,7 +263,7 @@
             'WORLD_SIZE': '2' if torch.cuda.device_count() == 2 else '3',
             'TEST_REPORT_SOURCE_OVERRIDE': 'dist-nccl'
         }
-    if not TEST_WITH_ROCM and dist.is_gloo_available():
+    if dist.is_gloo_available():
         DISTRIBUTED_TESTS_CONFIG['gloo'] = {
             'WORLD_SIZE': '2' if torch.cuda.device_count() == 2 else '3',
             'TEST_REPORT_SOURCE_OVERRIDE': 'dist-gloo'