Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/distributed/fsdp/
H A D_init_utils.py230 def _init_inter_node_process_group( function
281 _init_inter_node_process_group(global_process_group, num_devices_per_node),
/aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/
H A Dtest_fully_shard_init.py28 _init_inter_node_process_group,
869 dp_replicate_group = _init_inter_node_process_group(