diff --git a/imperative/python/megengine/distributed/__init__.py b/imperative/python/megengine/distributed/__init__.py index daf866af..79993da9 100644 --- a/imperative/python/megengine/distributed/__init__.py +++ b/imperative/python/megengine/distributed/__init__.py @@ -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 diff --git a/imperative/python/megengine/distributed/helper.py b/imperative/python/megengine/distributed/helper.py index 978c2854..81cfc77b 100644 --- a/imperative/python/megengine/distributed/helper.py +++ b/imperative/python/megengine/distributed/helper.py @@ -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))