PipeBarrier

功能说明

阻塞相同流水，具有数据依赖的相同流水之间需要插此同步。

函数原型

template <pipe_t pipe> __aicore__ inline void PipeBarrier();

参数说明

返回值

无

支持的型号

Atlas 训练系列产品

Atlas推理系列产品（Ascend 310P处理器）AI Core

Atlas A2训练系列产品/Atlas 800I A2推理产品

约束说明

Scalar流水之间的同步由硬件自动保证，调用PipeBarrier<PIPE_S>()会引发硬件错误。

调用示例

如下示例，Mul指令的输入dst0Local是Add指令的输出，两个矢量运算指令产生依赖，需要插入PipeBarrier保证两条指令的执行顺序。

图1 Mul指令和Add指令是串行关系，必须等待Add指令执行完成后，才能执行Mul指令。

LocalTensor<half> src0Local;
LocalTensor<half> src1Local;
LocalTensor<half> src2Local;
LocalTensor<half> dst0Local;
LocalTensor<half> dst1Local;

Add(dst0Local, src0Local, src1Local, 512);
PipeBarrier<PIPE_V>();
Mul(dst1Local, dst0Local, src2Local, 512);