Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_c10d_gloo.py1498 def _test_gloo_backend( member in DistributedDataParallelTest
1515 self._test_gloo_backend([torch.device("cpu")], None)
1519 self._test_gloo_backend(
1528 self._test_gloo_backend(devices, int_devices)
1535 self._test_gloo_backend(devices, devices)
1542 self._test_gloo_backend(devices, None, multi_device=True)
1549 self._test_gloo_backend(devices, None, multi_device=True)