Home
last modified time | relevance | path

Searched refs:MultiProcessTestCase (Results 1 – 25 of 28) sorted by relevance

12

/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_distributed.py521 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 Dtest_c10d_pypg.py14 from torch.testing._internal.common_distributed import MultiProcessTestCase
145 class TestDDPWithWorkSubclass(AbstractDDPSingleRank, MultiProcessTestCase):
151 class TestDDPWithWorkWrapper(AbstractDDPSingleRank, MultiProcessTestCase):
H A Dtest_collective_utils.py7 from torch.testing._internal.common_distributed import MultiProcessTestCase
10 class TestCollectiveUtils(MultiProcessTestCase):
H A Dtest_c10d_object_collectives.py15 from torch.testing._internal.common_distributed import MultiProcessTestCase, TEST_SKIPS
47 class TestObjectCollectives(MultiProcessTestCase):
H A Dtest_c10d_logger.py20 from torch.testing._internal.common_distributed import MultiProcessTestCase, TEST_SKIPS
52 class C10dErrorLoggerTest(MultiProcessTestCase):
H A Dtest_c10d_ucc.py35 MultiProcessTestCase,
145 class ProcessGroupUCCTest(MultiProcessTestCase):
343 test_c10d_common.CommonDistributedDataParallelTest, MultiProcessTestCase
978 class CommTest(test_c10d_common.AbstractCommTest, MultiProcessTestCase):
H A Dtest_c10d_spawn.py10 from torch.testing._internal.common_distributed import MultiProcessTestCase
104 class TestDistributedNNFunctions(MultiProcessTestCase):
H A Dtest_pg_wrapper.py21 MultiProcessTestCase,
30 class AbstractProcessGroupWrapperTest(MultiProcessTestCase):
H A Dtest_symmetric_memory.py17 MultiProcessTestCase,
63 class SymmetricMemoryTest(MultiProcessTestCase):
H A Dtest_c10d_nccl.py46 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 Dtest_functional_api.py25 MultiProcessTestCase,
462 class TestCollectivesWithNCCL(MultiProcessTestCase):
787 class TestFunctionalAutogradWithNCCL(MultiProcessTestCase):
H A Dtest_c10d_common.py32 MultiProcessTestCase,
1497 class CommTest(AbstractCommTest, MultiProcessTestCase):
1622 class PythonProcessGroupExtensionTest(MultiProcessTestCase):
1788 class ProcessGroupWithDispatchedCollectivesTests(MultiProcessTestCase):
2015 class LocalRankTest(MultiProcessTestCase):
H A Dtest_c10d_functional_native.py22 MultiProcessTestCase,
58 class TestWithNCCL(MultiProcessTestCase):
H A Dtest_c10d_gloo.py45 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__.py10 MultiProcessTestCase,
17 class ShardedTensorTestBase(MultiProcessTestCase):
/aosp_15_r20/external/pytorch/test/distributed/_composable/
H A Dtest_replicate.py14 MultiProcessTestCase,
31 class ReplicateStateDictTest(MultiProcessTestCase):
87 class ReplicateTest(MultiProcessTestCase):
H A Dtest_replicate_with_compiler.py30 MultiProcessTestCase,
74 class MultiProcessInductorTestCase(MultiProcessTestCase, InductorTestCase):
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Drpc_utils.py8 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 Dtest_ddp_hooks.py21 MultiProcessTestCase,
63 class DistributedDataParallelCommHookTest(MultiProcessTestCase):
/aosp_15_r20/external/pytorch/test/distributed/_composable/test_composability/
H A Dtest_pp_composability.py26 MultiProcessTestCase,
53 class ComposabilityTest(MultiProcessTestCase):
/aosp_15_r20/external/pytorch/test/distributed/algorithms/quantization/
H A Dtest_quantization.py13 MultiProcessTestCase,
57 class DistQuantizationTests(MultiProcessTestCase):
/aosp_15_r20/external/pytorch/test/distributed/algorithms/
H A Dtest_join.py18 MultiProcessTestCase,
137 class TestJoin(MultiProcessTestCase):
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/_tensor/
H A Dcommon_dtensor.py26 MultiProcessTestCase,
297 class DTensorTestBase(MultiProcessTestCase):
/aosp_15_r20/external/pytorch/test/distributed/checkpoint/
H A Dtest_state_dict.py50 MultiProcessTestCase,
861 class TestNoComm(MultiProcessTestCase):
/aosp_15_r20/external/pytorch/torch/distributed/
H A DCONTRIBUTING.md59 For an example of using the MultiProcessTestCase to run a test on multiple GPUs, see tests in [test…

12