Mask接口汇总【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit以下内容针对如下型号生效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时表示在接口内设置MaskisSetMask为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.md)-• 无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。排序组合ISASISort32、MrgSort-• 无isSetMask。• 接口不支持Mask。离散与聚合Gather、Gatherb.md)• 无isSetMask。• 接口不支持Mask。• 无isSetMask。• 接口不支持Mask。掩码操作SetMaskCount、SetMaskNorm、SetVectorMask、ResetMask--【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考