mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
779fc29c04
Pull Request resolved: https://github.com/pytorch/pytorch/pull/160892 Approved by: https://github.com/fduwjj
17 lines
450 B
Python
17 lines
450 B
Python
# Owner(s): ["oncall: distributed"]
|
|
|
|
from torch.testing._internal.common_distributed import MultiProcContinuousTest
|
|
from torch.testing._internal.common_utils import run_tests
|
|
|
|
|
|
class TestTemplate(MultiProcContinuousTest):
|
|
def testABC(self):
|
|
print(f"rank {self.rank} of {self.world_size} testing ABC")
|
|
|
|
def testDEF(self):
|
|
print(f"rank {self.rank} of {self.world_size} testing DEF")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_tests()
|