CANN/asc-devkit带进位加法API

AddC【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√Atlas A3 训练系列产品 / Atlas A3 推理系列产品xAtlas A2 训练系列产品 / Atlas A2 推理系列产品x功能说明根据mask对输入数据src0、src1以及进位数据carrySrc进行按元素相加操作将结果写入dst。如果src0, src1输入转换为uint32_t类型加上进位值carrySrc相加时超出uint32_t最大值在MaskReg carry中对应位置每4bit大小写1否则写0。计算公式如下对carry的操作示例如下int32_t类型a_i, b_i∈[-2147483648, 2147483647]假设a_i -2147483648, b_i -2, carrySrc_i 1(uint32_t)a_i (uint32_t)b_i (uint32_t)carrySrc_i (uint64_t)uint_dst_i因为uint_dst_i 32 大于0所以carry[(i%64):4*(i%64)] 1假设a_i 2, b_i 5, carrySrc_i 1(uint32_t)a_i (uint32_t)b_i (uint32_t)carrySrc_i (uint64_t)uint_dst_i因为uint_dst_i 32 等于0所以carry[(i%64):4*(i%64)] 0uint32_t类型a_i, b_i∈[0, 4294967295]假设a_i 4294967295, b_i 2, carrySrc_i 0(uint32_t)a_i (uint32_t)b_i (uint32_t)carrySrc_i (uint64_t)uint_dst_i因为uint_dst_i 32 大于0所以carry[(i%64):4*(i%64)] 1假设a_i 3, b_i 2, carrySrc_i 0(uint32_t)a_i (uint32_t)b_i (uint32_t)carrySrc_i (uint64_t)uint_dst_i因为uint_dst_i 32 等于0所以carry[(i%64):4*(i%64)] 0函数原型template typename T DefaultType, typename U __simd_callee__ inline void AddC(MaskReg carry, U dstReg, U srcReg0, U srcReg1, MaskReg carrySrc, MaskReg mask)参数说明表 1模板参数说明参数名描述T操作数数据类型。Ascend 950PR/Ascend 950DT支持的数据类型为uint32_t/int32_t。U操作数RegTensor类型 例如RegTensoruint32_t由编译器自动推导用户不需要填写。表 2参数说明参数名输入/输出描述carry输出目的操作数类型为MaskReg。dstReg输出目的操作数。类型为RegTensor。srcReg0输入源操作数。类型为RegTensor。三个源操作数的数据类型需要与目的操作数保持一致。srcReg1输入源操作数。类型为RegTensor。三个源操作数的数据类型需要与目的操作数保持一致。carrySrc输入源操作数。输入进位值。类型为MaskReg。三个源操作数的数据类型需要与目的操作数保持一致。Ascend 950PR/Ascend 950DT支持的数据类型为uint32_t/int32_t。mask输入源操作数元素操作的有效指示详细说明请参考MaskReg。返回值说明无约束说明无调用示例template typename T static __simd_vf__ inline void AddVF(__ubuf__ T* dst0Addr, __ubuf__ T* dst1Addr, __ubuf__ T* src0Addr, __ubuf__ T* src1Addr, uint32_t count, uint16_t repeatTimes, uint32_t oneRepeatSize){ AscendC::Reg::RegTensorT srcReg0; AscendC::Reg::RegTensorT srcReg1; AscendC::Reg::RegTensorT dstReg0; AscendC::Reg::RegTensorT dstReg1; AscendC::Reg::MaskReg mask; AscendC::Reg::MaskReg carry AscendC::Reg::CreateMaskuint8_t(); for (uint16_t i 0; i repeatTimes; i) { mask AscendC::Reg::UpdateMaskT(count); AscendC::Reg::LoadAlign(srcReg0, src0Addr i * oneRepeatSize); AscendC::Reg::LoadAlign(srcReg1, src1Addr i * oneRepeatSize); AscendC::Reg::Add(carry, dstReg0, srcReg0, srcReg1, mask); // 8*4B32B align AscendC::Reg::StoreAlignuint32_t, AscendC::Reg::MaskDist::DIST_NORM((__ubuf__ uint32_t*)dst1Addr i * 8, carry); AscendC::Reg::StoreAlign(dst0Addr i * oneRepeatSize, dstReg0, mask); } }【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考