Searched defs:_create_wrapper_pg (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/ | ||
| H A D | test_pg_wrapper.py | 224 def _create_wrapper_pg(self, with_new_group=False, timeout=10.0): member in ProcessGroupNCCLWrapperTest |