MCPcopy Create free account
hub / github.com/pytorch/pytorch / _SyncAllParamsSingleHost

Function _SyncAllParamsSingleHost

caffe2/python/data_parallel_model.py:1232–1234  ·  view source on GitHub ↗
(devices, model, net, unique_param_names)

Source from the content-addressed store, hash-verified

1230
1231
1232def _SyncAllParamsSingleHost(devices, model, net, unique_param_names):
1233 for param in unique_param_names:
1234 _Broadcast(devices, model, net, param)
1235
1236
1237def _AllReduceBlobs(blob_names, devices, model, net, rendezvous, use_nccl,

Callers 1

_SyncAllParamsFunction · 0.85

Calls 1

_BroadcastFunction · 0.70

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…