From f1c985695c97d811185f6f1527398e1fd87ed028 Mon Sep 17 00:00:00 2001 From: Jithun Nair Date: Wed, 25 Nov 2020 19:50:30 -0800 Subject: [PATCH] Enabled gloo backend in test_distributed unit tests for ROCm (#40395) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/40395 Reviewed By: ngimel Differential Revision: D25181692 Pulled By: mrshenli fbshipit-source-id: 29f478c974791efc0acea210c8c9e574944746a5 --- test/run_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/run_test.py b/test/run_test.py index 6423c2eba3e..8653489ec63 100755 --- a/test/run_test.py +++ b/test/run_test.py @@ -263,7 +263,7 @@ if dist.is_available(): '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'