Function _BroadcastComputedParamsDistributed

caffe2/python/data_parallel_model.py:1522–1529

(
    devices,
    model,
    rendezvous,
    use_nccl=False
)

def _BroadcastComputedParamsDistributed(
    devices,
    model,
    rendezvous,
    use_nccl=False
):
    _BroadcastComputedParamsSingleHost(devices, model, use_nccl)
    log.warn("Distributed broadcast of computed params is not implemented yet")


def _BroadcastComputedParamsSingleHost(devices, model, use_nccl=False):