saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

Browse source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/40395

Reviewed By: ngimel

Differential Revision: D25181692

Pulled By: mrshenli

fbshipit-source-id: 29f478c974791efc0acea210c8c9e574944746a5
This commit is contained in:
Jithun Nair 2020-11-25 19:50:30 -08:00 committed by Facebook GitHub Bot
parent db1b0b06c4
commit f1c985695c
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
test/run_test.py
View file

@ -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'