Home
last modified time | relevance | path

Searched refs:MultiThreadedTestCase (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_functional_api.py26 MultiThreadedTestCase,
60 class TestExpand(MultiThreadedTestCase):
149 class TestPgTag(MultiThreadedTestCase):
225 class TestTraceableCollectives(MultiThreadedTestCase):
370 class TestGradCollectives(MultiThreadedTestCase):
654 class TestFunctionalAutograd(MultiThreadedTestCase):
H A Dtest_multi_threaded_pg.py21 MultiThreadedTestCase,
136 class TestCollectivesWithBaseClass(MultiThreadedTestCase):
H A Dtest_store.py19 from torch.testing._internal.common_distributed import MultiThreadedTestCase
873 class TestMultiThreadedWait(MultiThreadedTestCase):
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_distributed.py947 MultiThreadedTestCase.exception_queue.put((rank, sys.exc_info()))
969 MultiThreadedTestCase._join_threads(threads, func)
976 class MultiThreadedTestCase(TestCase): class
1102 MultiThreadedTestCase.exception_queue.put(
H A Dcommon_fsdp.py59 MultiThreadedTestCase,
1091 class FSDPTestMultiThread(MultiThreadedTestCase):
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/_tensor/
H A Dcommon_dtensor.py27 MultiThreadedTestCase,
385 class DTensorOpTestBase(MultiThreadedTestCase):