Mask接口汇总
以下内容针对如下型号生效：
- Atlas A3 训练系列产品/Atlas A3 推理系列产品
- Atlas A2 训练系列产品/Atlas A2 推理系列产品
本节汇总了各接口中模板参数isSetMask的支持情况。
针对Tensor前n个数计算和Tensor高维切分计算两种接口，Mask设置方式分为以下情形：
isSetMask参数情况：
模板参数有isSetMask，支持接口外设置Mask。
模板参数无isSetMask，不支持接口外设置Mask。
模板参数有isSetMask，该参数不起作用。
Mask设置方式：
接口内已设置Mask：接口内部调用SetMaskCount、SetVectorMask、SetMaskNorm、ResetMask。
接口内通过模板参数控制：接口内通过isSetMask模板参数控制如何设置Mask。isSetMask为true时表示在接口内设置Mask；isSetMask为false时表示在接口外设置Mask，开发者需使用SetVectorMask接口设置Mask值。
接口不支持Mask或者无需设置Mask。
更多Mask信息可参考基于全局掩码复用的计算性能优化。
注：表中的-表示不存在该类型接口。
表 1 各接口模板参数isSetMask支持情况汇总表
| 接口类别 | 接口名称 | Tensor前n个数计算 | Tensor高维切分计算 |
|---|---|---|---|
| 基础算术 | Add、Exp、Ln、Abs、Reciprocal、Sqrt、Rsqrt、Relu、Sub、Mul、Div、Max、Min | • 无isSetMask。 • 接口内已设置Mask。 | • 有isSetMask。 • 接口内通过模板参数控制。 |
| 基础算术 | BilinearInterpolation | - | • 无isSetMask。 • 接口内已设置Mask。 |
| 基础算术 | Adds、Muls、Maxs、Mins、LeakyRelu | • 有isSetMask。 • 接口内已设置Mask。 | • 有isSetMask。 • 接口内通过模板参数控制。 |
| 逻辑计算 | Not、And、Or | • 无isSetMask。 • 接口内已设置Mask。 | • 有isSetMask。 • 接口内通过模板参数控制。 |
| 逻辑计算 | ShiftLeft、ShiftRight | • 有isSetMask。 • 接口内通过模板参数控制。 | • 有isSetMask。 • 接口内通过模板参数控制。 |
| 复合计算 | Axpy、MulAddDst、MulCast、AddRelu、AddReluCast、SubRelu、SubReluCast、FusedMulAdd、MulAddRelu、CastDequant、AddDeqRelu | • 无isSetMask。 • 接口内已设置Mask。 | • 有isSetMask。 • 接口内通过模板参数控制。 |
| 比较与选择 | Compare | • 无isSetMask。 • 接口不支持Mask。 | • 有isSetMask，不起作用。 • 接口无需设置Mask。 |
| 比较与选择 | Compare（结果存入寄存器） | - | • 有isSetMask。 • 接口内通过模板参数控制。 |
| 比较与选择 | Compares | • 无isSetMask。 • 接口不支持Mask。 | • 有isSetMask。 • 接口无需设置Mask。 |
| 比较与选择 | Select | • 无isSetMask。 • 接口内已设置Mask。 | • 有isSetMask，不起作用。 • 如需使用在接口外部设置Mask的功能，可以调用不传入mask参数的接口来实现。 |
| 比较与选择 | GatherMask | - | • 无isSetMask。 • 接口内已设置Mask。 |
| 类型转换 | Cast | • 无isSetMask。 • 接口内已设置Mask。 | • 有isSetMask。 • 接口内通过模板参数控制。 |
| 归约计算 | ReduceDataBlock、ReduceRepeat、ReducePairElem | - | • 有isSetMask。 • 接口内通过模板参数控制。 |
| 归约计算 | ReduceSum | • 有isSetMask。 • 接口内通过模板参数控制。 | • 无isSetMask。 • 接口内已设置Mask。 |
| 归约计算 | ReduceMax、ReduceMin | • 无isSetMask。 • 接口内已设置Mask。 | • 无isSetMask。 • 接口内已设置Mask。 |
| 数据排布转换 | Transpose、TransDataTo5HD | • 无isSetMask。 • 接口不支持Mask。 | • 无isSetMask。 • 接口不支持Mask。 |
| 数据填充 | Brcb | - | • 无isSetMask。 • 接口不支持Mask。 |
| 数据填充 | Duplicate | • 无isSetMask。 • 接口内已设置Mask。 | • 有isSetMask。 • 接口内通过模板参数控制。 |
| 数据填充 | CreateVecIndex | • 无isSetMask。 • 接口不支持Mask。 | • 无isSetMask。 • 接口不支持Mask。 |
| 排序组合（ISASI） | Sort32、MrgSort | - | • 无isSetMask。 • 接口不支持Mask。 |
| 离散与聚合 | Gather、Gatherb | • 无isSetMask。 • 接口不支持Mask。 | • 无isSetMask。 • 接口不支持Mask。 |
| 掩码操作 | SetMaskCount、SetMaskNorm、SetVectorMask、ResetMask | - | - |