Searched refs:ModuleForDdpCommHook (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_c10d_ucc.py | 24 ModuleForDdpCommHook,
804 ModuleForDdpCommHook().cpu(), process_group=process_group
819 ModuleForDdpCommHook().to(device_id),
857 ModuleForDdpCommHook(), process_group=process_group
884 ModuleForDdpCommHook(), process_group=process_group
927 ModuleForDdpCommHook(), process_group=process_group
|
| H A D | test_c10d_gloo.py | 27 ModuleForDdpCommHook,
1976 ModuleForDdpCommHook().cpu(), process_group=process_group
1991 ModuleForDdpCommHook().to(device_id),
2029 ModuleForDdpCommHook(), process_group=process_group
2056 ModuleForDdpCommHook(), process_group=process_group
2099 ModuleForDdpCommHook(), process_group=process_group
|
| H A D | test_c10d_common.py | 301 class ModuleForDdpCommHook(nn.Module): class
833 ModuleForDdpCommHook().to(device_id),
850 ModuleForDdpCommHook().to(device_id),
|
| H A D | test_c10d_nccl.py | 32 from test_c10d_common import ConvNet, DoubleGpuNet, gpus_for_rank, ModuleForDdpCommHook
1753 ModuleForDdpCommHook().to(device_id),
|