SetMaskCount

功能说明

设置mask模式为Counter模式。该模式下，不需要开发者去感知迭代次数、处理非对齐的尾块等操作，可直接传入计算数据量，实际迭代次数由Vector计算单元自动推断。mask模式分为Counter模式和Normal模式，两种模式的概念和使用场景请参考掩码操作。Counter模式的设置和使用流程请参考如何手动设置Counter模式。

函数原型

1

__aicore__ inline void SetMaskCount()

参数说明

无。

返回值

无。

支持的型号

Atlas 推理系列产品 AI Core

Atlas 推理系列产品 Vector Core

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

Atlas 200I/500 A2 推理产品

约束说明

设置为Counter模式的场景需要在矢量计算使用完之后调用SetMaskNorm将mask模式恢复为Normal模式。

调用示例

请参考Counter模式调用示例。