Searched refs:_init_intra_node_process_group (Results 1 – 2 of 2) sorted by relevance
214 def _init_intra_node_process_group(num_devices_per_node: int) -> dist.ProcessGroup: function280 _init_intra_node_process_group(num_devices_per_node),
29 _init_intra_node_process_group,868 dp_shard_group = _init_intra_node_process_group(shard_mesh_dim_size)