Browse Source

refactor(mge/distributed): change bcast_params_ to bcast_list_

GitOrigin-RevId: 26b452a6b7
tags/v1.0.0-rc1
Megvii Engine Team 4 years ago
parent
commit
8d02d10483
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      imperative/python/megengine/distributed/__init__.py
  2. +1
    -1
      imperative/python/megengine/distributed/helper.py

+ 1
- 1
imperative/python/megengine/distributed/__init__.py View File

@@ -19,7 +19,7 @@ from .group import (
is_distributed,
new_group,
)
from .helper import bcast_params_, make_allreduce_cb, synchronized
from .helper import bcast_list_, make_allreduce_cb, synchronized
from .launcher import launcher
from .server import Client, Server
from .util import get_free_ports

+ 1
- 1
imperative/python/megengine/distributed/helper.py View File

@@ -67,7 +67,7 @@ def get_device_count_by_fork(device_type: str):
return q.get()


def bcast_params_(params, group):
def bcast_list_(params, group):
for p in params:
p._reset(broadcast(p, group))



Loading…
Cancel
Save