提交 8d02d104 编写于 作者: M Megvii Engine Team

refactor(mge/distributed): change bcast_params_ to bcast_list_

GitOrigin-RevId: 26b452a6b763c7eac621e00524198db6a3b558c1
上级 c7acba41
......@@ -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
......@@ -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))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册