CANN/asc-devkit SIMD向量长度获取函数
GetVecLen【免费下载链接】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 推理系列产品xAtlas 200I/500 A2 推理产品xAtlas 推理系列产品 AI CorexAtlas 推理系列产品 Vector CorexAtlas 训练系列产品xKirin X90xKirin 9030x功能说明获取RegTensor位宽VLVector Length的大小。函数原型__aicore__ inline constexpr uint32_t GetVecLen()参数说明无返回值说明Vector Length的大小单位为byte。约束说明无调用示例如下样例通过GetVecLen获取循环迭代次数template typename T __aicore__ inline void AddCustomImpl(__local_mem__ T *dst, __local_mem__ T *src0, __local_mem__ T *src1, uint32_t calCount) { AscendC::Reg::RegTensorT reg0; AscendC::Reg::RegTensorT reg1; AscendC::Reg::RegTensorT reg2; AscendC::Reg::MaskReg mask; constexpr uint32_t repeatElm AscendC::GetVecLen() / sizeof(T); uint16_t repeatTime AscendC::CeilDivision(calCount, repeatElm); for (uint16_t i 0; i repeatTime; i) { mask AscendC::Reg::UpdateMaskT(calCount); AscendC::Reg::LoadAlign(reg0, src0 i * repeatElm); AscendC::Reg::LoadAlign(reg1, src1 i * repeatElm); AscendC::Reg::Add(reg2, reg0, reg1, mask); AscendC::Reg::StoreAlign(dst i * repeatElm, reg2, mask); } }【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2629186.html
如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!