Home
last modified time | relevance | path

Searched refs:CommonDistributedDataParallelTest (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_c10d_pypg.py94 class AbstractDDPSingleRank(test_c10d_common.CommonDistributedDataParallelTest):
H A Dtest_c10d_common.py319 class CommonDistributedDataParallelTest: class
988 return CommonDistributedDataParallelTest.CustomOutput(o1=o1, o2=o2)
1785 instantiate_parametrized_tests(CommonDistributedDataParallelTest)
H A Dtest_c10d_ucc.py343 test_c10d_common.CommonDistributedDataParallelTest, MultiProcessTestCase
H A Dtest_c10d_gloo.py1485 test_c10d_common.CommonDistributedDataParallelTest, MultiProcessTestCase
H A Dtest_c10d_nccl.py845 test_c10d_common.CommonDistributedDataParallelTest, MultiProcessTestCase