/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…
|