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

docs(mge/distributed): add document for distributed.backend

GitOrigin-RevId: 6cdcf7af7796e551a3bbaca9f84531fd84791db8
上级 9526ee52
...@@ -31,6 +31,18 @@ from .server import Client, Server ...@@ -31,6 +31,18 @@ from .server import Client, Server
@mproperty @mproperty
def backend(mod): def backend(mod):
r"""
Get or set backend of collective communication.
Available backends are ['nccl', 'shm', 'rccl']
Examples:
.. code-block::
import megengine.distributed as dist
dist.backend = "nccl"
"""
assert group._sd, "please call init_process_group first" assert group._sd, "please call init_process_group first"
return group._sd.backend return group._sd.backend
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册