commit ef50c94e7ccb67129935b5c64747f2a5645790f8
author Rong Rong <rongr@fb.com> Thu Dec 03 06:48:51 2020 -0800
committer Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> Thu Dec 03 06:50:36 2020 -0800
tree dbe4a185f8436237073f42d8661b32df687b0a96
parent 0484b048d050ea5b10cb0efd147148aa893f2a4b

reenabling MPI test (#48725)

Summary:
fixes https://github.com/pytorch/pytorch/issues/47443.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/48725

Reviewed By: mrshenli

Differential Revision: D25278758

Pulled By: walterddr

fbshipit-source-id: a02d0fef99a7941c8e98da16a45d840e12b8b0c3

test/distributed/test_distributed_fork.py

diff --git a/test/distributed/test_distributed_fork.py b/test/distributed/test_distributed_fork.py
index 293eba1..84d23e7 100644
--- a/test/distributed/test_distributed_fork.py
+++ b/test/distributed/test_distributed_fork.py
@@ -54,7 +54,7 @@
     WORLD_SIZE = os.environ["WORLD_SIZE"]
     dist.init_process_group(init_method=INIT_METHOD, backend="mpi")
 
-    class TestMPI(DistributedTest._DistTestBase):
+    class TestMPIWithFork(TestCase, DistributedTest._DistTestBase):
         pass
 
 elif BACKEND == "test":

test/run_test.py

diff --git a/test/run_test.py b/test/run_test.py
index 070b610..3687459 100755
--- a/test/run_test.py
+++ b/test/run_test.py
@@ -415,7 +415,7 @@
                 init_str = "with {} init_method"
                 with_init = init_str.format("file" if with_init_file else "env")
                 print_to_stderr(
-                    'Running distributed tests for the {} backend{}'.format(
+                    'Running distributed tests for the {} backend {}'.format(
                         backend, with_init))
             os.environ['TEMP_DIR'] = tmp_dir
             os.environ['BACKEND'] = backend