SetMaskNorm
功能说明
设置mask模式为Normal模式。该模式为系统默认模式,支持开发者配置迭代次数,通用说明中介绍的配置方式为Normal模式。mask模式分为Counter模式和Normal模式,两种模式的概念和使用场景请参考掩码操作。
Normal模式的设置和使用流程如下:
- 设置mask的模式为Normal后,调用SetVectorMask指令来设置mask,此时的mask用来控制单次迭代内参与计算的元素个数。
- mask设置完成,即可调用矢量计算高维切分计算接口完成计算。
- 矢量计算接口的isSetMask模板参数需要设置为false,表示在矢量计算接口外部设置mask;接口入参中的mask值设置为MASK_PLACEHOLDER,用于占位,无实际含义。
- 根据使用场景正确配置repeatTimes、dataBlockStride、repeatStride参数。
- 使用ResetMask恢复mask的值为默认值。
函数原型
1
|
__aicore__ inline void SetMaskNorm() |
参数说明
无。
返回值
无。
支持的型号
Atlas推理系列产品AI Core
Atlas推理系列产品Vector Core
Atlas A2训练系列产品/Atlas 800I A2推理产品
Atlas 200/500 A2推理产品
约束说明
无。
调用示例
请参考Normal模式调用示例。
父主题: 掩码操作