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 200/500 A2推理产品
约束说明
设置为Counter模式的场景需要在矢量计算使用完之后调用SetMaskNorm将mask模式恢复为Normal模式。
调用示例
请参考Counter模式调用示例。
父主题: 掩码操作