| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_distributed.py | 521 class MultiProcessTestCase(TestCase): class
624 if event == MultiProcessTestCase.Event.GET_TRACEBACK:
649 target=MultiProcessTestCase._event_listener,
674 traceback.format_exc(), self.rank, MultiProcessTestCase.TEST_ERROR_EXIT_CODE
678 sys.exit(MultiProcessTestCase.TEST_ERROR_EXIT_CODE)
694 pipe.send(MultiProcessTestCase.Event.GET_TRACEBACK)
735 if p.exitcode == MultiProcessTestCase.TEST_ERROR_EXIT_CODE:
804 if p.exitcode == MultiProcessTestCase.TEST_ERROR_EXIT_CODE
1290 class DynamoDistributedMultiProcTestCase(MultiProcessTestCase):
|
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_c10d_pypg.py | 14 from torch.testing._internal.common_distributed import MultiProcessTestCase
145 class TestDDPWithWorkSubclass(AbstractDDPSingleRank, MultiProcessTestCase):
151 class TestDDPWithWorkWrapper(AbstractDDPSingleRank, MultiProcessTestCase):
|
| H A D | test_collective_utils.py | 7 from torch.testing._internal.common_distributed import MultiProcessTestCase
10 class TestCollectiveUtils(MultiProcessTestCase):
|
| H A D | test_c10d_object_collectives.py | 15 from torch.testing._internal.common_distributed import MultiProcessTestCase, TEST_SKIPS
47 class TestObjectCollectives(MultiProcessTestCase):
|
| H A D | test_c10d_logger.py | 20 from torch.testing._internal.common_distributed import MultiProcessTestCase, TEST_SKIPS
52 class C10dErrorLoggerTest(MultiProcessTestCase):
|
| H A D | test_c10d_ucc.py | 35 MultiProcessTestCase,
145 class ProcessGroupUCCTest(MultiProcessTestCase):
343 test_c10d_common.CommonDistributedDataParallelTest, MultiProcessTestCase
978 class CommTest(test_c10d_common.AbstractCommTest, MultiProcessTestCase):
|
| H A D | test_c10d_spawn.py | 10 from torch.testing._internal.common_distributed import MultiProcessTestCase
104 class TestDistributedNNFunctions(MultiProcessTestCase):
|
| H A D | test_pg_wrapper.py | 21 MultiProcessTestCase,
30 class AbstractProcessGroupWrapperTest(MultiProcessTestCase):
|
| H A D | test_symmetric_memory.py | 17 MultiProcessTestCase,
63 class SymmetricMemoryTest(MultiProcessTestCase):
|
| H A D | test_c10d_nccl.py | 46 MultiProcessTestCase,
212 class ProcessGroupNCCLGroupTest(MultiProcessTestCase):
845 test_c10d_common.CommonDistributedDataParallelTest, MultiProcessTestCase
2211 class WorkHookTest(MultiProcessTestCase):
2434 class NcclErrorHandlingTest(MultiProcessTestCase):
2652 class CommTest(test_c10d_common.AbstractCommTest, MultiProcessTestCase):
3171 class LargeCommTest(test_c10d_common.AbstractLargeCommTest, MultiProcessTestCase):
3446 class SparseCollective(MultiProcessTestCase):
3520 class NCCLTraceTestBase(MultiProcessTestCase):
4162 return MultiProcessTestCase._check_return_codes(self, *args, **kwargs)
[all …]
|
| H A D | test_functional_api.py | 25 MultiProcessTestCase,
462 class TestCollectivesWithNCCL(MultiProcessTestCase):
787 class TestFunctionalAutogradWithNCCL(MultiProcessTestCase):
|
| H A D | test_c10d_common.py | 32 MultiProcessTestCase,
1497 class CommTest(AbstractCommTest, MultiProcessTestCase):
1622 class PythonProcessGroupExtensionTest(MultiProcessTestCase):
1788 class ProcessGroupWithDispatchedCollectivesTests(MultiProcessTestCase):
2015 class LocalRankTest(MultiProcessTestCase):
|
| H A D | test_c10d_functional_native.py | 22 MultiProcessTestCase,
58 class TestWithNCCL(MultiProcessTestCase):
|
| H A D | test_c10d_gloo.py | 45 MultiProcessTestCase,
214 class ProcessGroupGlooTest(MultiProcessTestCase):
1485 test_c10d_common.CommonDistributedDataParallelTest, MultiProcessTestCase
2302 class CommTest(test_c10d_common.AbstractCommTest, MultiProcessTestCase):
2466 class LargeCommTest(test_c10d_common.AbstractLargeCommTest, MultiProcessTestCase):
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/_shard/sharded_tensor/ |
| H A D | __init__.py | 10 MultiProcessTestCase,
17 class ShardedTensorTestBase(MultiProcessTestCase):
|
| /aosp_15_r20/external/pytorch/test/distributed/_composable/ |
| H A D | test_replicate.py | 14 MultiProcessTestCase,
31 class ReplicateStateDictTest(MultiProcessTestCase):
87 class ReplicateTest(MultiProcessTestCase):
|
| H A D | test_replicate_with_compiler.py | 30 MultiProcessTestCase,
74 class MultiProcessInductorTestCase(MultiProcessTestCase, InductorTestCase):
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | rpc_utils.py | 8 from torch.testing._internal.common_distributed import MultiProcessTestCase
87 class SpawnHelper(MultiProcessTestCase):
|
| /aosp_15_r20/external/pytorch/test/distributed/algorithms/ddp_comm_hooks/ |
| H A D | test_ddp_hooks.py | 21 MultiProcessTestCase,
63 class DistributedDataParallelCommHookTest(MultiProcessTestCase):
|
| /aosp_15_r20/external/pytorch/test/distributed/_composable/test_composability/ |
| H A D | test_pp_composability.py | 26 MultiProcessTestCase,
53 class ComposabilityTest(MultiProcessTestCase):
|
| /aosp_15_r20/external/pytorch/test/distributed/algorithms/quantization/ |
| H A D | test_quantization.py | 13 MultiProcessTestCase,
57 class DistQuantizationTests(MultiProcessTestCase):
|
| /aosp_15_r20/external/pytorch/test/distributed/algorithms/ |
| H A D | test_join.py | 18 MultiProcessTestCase,
137 class TestJoin(MultiProcessTestCase):
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/_tensor/ |
| H A D | common_dtensor.py | 26 MultiProcessTestCase,
297 class DTensorTestBase(MultiProcessTestCase):
|
| /aosp_15_r20/external/pytorch/test/distributed/checkpoint/ |
| H A D | test_state_dict.py | 50 MultiProcessTestCase,
861 class TestNoComm(MultiProcessTestCase):
|
| /aosp_15_r20/external/pytorch/torch/distributed/ |
| H A D | CONTRIBUTING.md | 59 For an example of using the MultiProcessTestCase to run a test on multiple GPUs, see tests in [test…
|