From 8d02d10483b0c30ac31c52d6ec47cc5f1fc92c5d Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Wed, 9 Sep 2020 18:27:52 +0800 Subject: [PATCH] refactor(mge/distributed): change bcast_params_ to bcast_list_ GitOrigin-RevId: 26b452a6b763c7eac621e00524198db6a3b558c1 --- imperative/python/megengine/distributed/__init__.py | 2 +- imperative/python/megengine/distributed/helper.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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))