OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[BE] rename some tests in test_c10d_common (#67828)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/67828

as titled
ghstack-source-id: 143781976

Test Plan: wait for ci

Reviewed By: mrshenli

Differential Revision: D32165576

fbshipit-source-id: 40c04b74f9e3241d3b3d64dee53af01fcfd1018b
This commit is contained in:
Wanchao Liang 2021-11-18 17:13:02 -08:00 committed by Facebook GitHub Bot
parent 3757a16c7a
commit 003f6ccec6
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
test/distributed/test_c10d_common.py
View File

@ -792,13 +792,13 @@ class DummyProcessGroup(dist.ProcessGroup):
return DummyWork() return DummyWork()
class PythonProcessGroupTest(MultiProcessTestCase): class PythonProcessGroupExtensionTest(MultiProcessTestCase):
def setUp(self): def setUp(self):
super(PythonProcessGroupTest, self).setUp() super(PythonProcessGroupExtensionTest, self).setUp()
self._spawn_processes() self._spawn_processes()
def tearDown(self): def tearDown(self):
super(PythonProcessGroupTest, self).tearDown() super(PythonProcessGroupExtensionTest, self).tearDown()
try: try:
os.remove(self.file_name) os.remove(self.file_name)
except OSError: except OSError:
@ -811,12 +811,12 @@ class PythonProcessGroupTest(MultiProcessTestCase):
def test_backend_class_attr(self): def test_backend_class_attr(self):
dist.Backend.register_backend( dist.Backend.register_backend(
"dummy", "dummy",
PythonProcessGroupTest.create_dummy PythonProcessGroupExtensionTest.create_dummy
) )
self.assertEqual(dist.Backend.DUMMY, "DUMMY") self.assertEqual(dist.Backend.DUMMY, "DUMMY")
self.assertEqual( self.assertEqual(
dist.Backend._plugins["DUMMY"], dist.Backend._plugins["DUMMY"],
PythonProcessGroupTest.create_dummy PythonProcessGroupExtensionTest.create_dummy
) )
@staticmethod @staticmethod
@ -824,7 +824,7 @@ class PythonProcessGroupTest(MultiProcessTestCase):
return DummyProcessGroup(rank, size) return DummyProcessGroup(rank, size)
def test_collectives(self): def test_collectives(self):
dist.Backend.register_backend("dummy", PythonProcessGroupTest.create_dummy) dist.Backend.register_backend("dummy", PythonProcessGroupExtensionTest.create_dummy)
os.environ['MASTER_ADDR'] = 'localhost' os.environ['MASTER_ADDR'] = 'localhost'
os.environ['MASTER_PORT'] = '6789' os.environ['MASTER_PORT'] = '6789'
@ -858,7 +858,7 @@ class PythonProcessGroupTest(MultiProcessTestCase):
dist.destroy_process_group() dist.destroy_process_group()
def test_send_recv(self): def test_send_recv(self):
dist.Backend.register_backend("dummy", PythonProcessGroupTest.create_dummy) dist.Backend.register_backend("dummy", PythonProcessGroupExtensionTest.create_dummy)
os.environ['MASTER_ADDR'] = 'localhost' os.environ['MASTER_ADDR'] = 'localhost'
os.environ['MASTER_PORT'] = '6789' os.environ['MASTER_PORT'] = '6789'