(self, parallel_mode: ParallelMode)
| 357 | return self._ranks_in_group[parallel_mode] |
| 358 | |
| 359 | def get_cpu_group(self, parallel_mode: ParallelMode): |
| 360 | self._check_parallel_mode(parallel_mode) |
| 361 | return self._cpu_groups[parallel_mode] |
| 362 | |
| 363 | def init_global_dist(self, rank: int, world_size: int, backend: str, host: str, port: int, use_cpu: bool = False): |
| 364 | """Initializes the global distributed environment |
no test coverage detected