未验证 提交 f9165878 编写于 作者: H houj04 提交者: GitHub

[XPU] fleet collective support bkcl. (#44929)

上级 d99cb2e1
...@@ -476,6 +476,10 @@ def new_group(ranks=None, backend=None): ...@@ -476,6 +476,10 @@ def new_group(ranks=None, backend=None):
place = core.MLUPlace(genv.device_id) place = core.MLUPlace(genv.device_id)
core.CNCLParallelContext(strategy, core.CNCLParallelContext(strategy,
place).init_with_ring_id(ring_id) place).init_with_ring_id(ring_id)
elif core.is_compiled_with_xpu():
place = core.XPUPlace(genv.device_id)
core.BKCLParallelContext(strategy,
place).init_with_ring_id(ring_id)
else: else:
assert False, ("no cuda device found") assert False, ("no cuda device found")
else: else:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册