2.6K Star 8.6K Fork 4.8K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.ops.Barrier.rst 1001 Bytes
一键复制 编辑 原始数据 按行查看 历史
horcham 提交于 2024-06-05 18:25 . fix docs for comm ops

mindspore.ops.Barrier

.. py:class:: mindspore.ops.Barrier(group=GlobalComm.WORLD_COMM_GROUP)

    同步通信域内的多个进程。进程调用到该算子后进入阻塞状态,直到通信域内所有进程调用到该算子,
    进程被唤醒并继续执行。

    参数:
        - **group** (str,可选) - 表示通信域。默认值: ``GlobalComm.WORLD_COMM_GROUP`` 。

    异常:
        - **TypeError** - 首个输入的数据类型不为Tensor,`op` 或 `group` 不是str。
        - **RuntimeError** - 后端无效,或者分布式初始化失败。
        - **ValueError** - 调用进程的rank id大于本通信组的rank大小。

    样例:

    .. note::
        .. include:: mindspore.ops.comm_note.rst

        该样例需要在4卡环境下运行。

    教程样例:
        - `分布式集合通信原语 - Barrier
          <https://www.mindspore.cn/docs/zh-CN/master/api_python/samples/ops/communicate_ops.html#barrier>`_
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
r2.3.1

搜索帮助