mindspore.ops.Barrier
- 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大小。
- 支持平台:
Ascend
样例:
>>> import numpy as np >>> import mindspore.nn as nn >>> from mindspore.communication import init >>> from mindspore import Tensor >>> from mindspore import ops >>> # Launch 4 processes. >>> init() >>> class BarrierNet(nn.Cell): >>> def __init__(self): >>> super(BarrierNet, self).__init__() >>> self.barrier = ops.Barrier() >>> >>> def construct(self): >>> self.barrier() >>> net = BarrierNet() >>> net()
- 教程样例: