Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/distributed/fsdp/
H A D_init_utils.py214 def _init_intra_node_process_group(num_devices_per_node: int) -> dist.ProcessGroup: function
280 _init_intra_node_process_group(num_devices_per_node),
/aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/
H A Dtest_fully_shard_init.py29 _init_intra_node_process_group,
868 dp_shard_group = _init_intra_node_process_group(shard_mesh_dim_size)