CANN/asc-devkit 向量计算类API样例介绍

发布时间:2026/6/13 16:19:02
CANN/asc-devkit 向量计算类API样例介绍 向量计算类api样例介绍【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit概述本路径下包含了与向量计算相关的多个API的样例。每个样例均基于Ascend C的直调方法支持main函数和kernel函数在同一个cpp文件中实现。算子开发样例目录名称功能描述abs本样例基于Reg编程接口实现Abs运算Relu/Exp/Sqrt/Ln/Log/Log2/Log10/Neg接口皆可参考该样例arange本样例基于Reg编程接口实现Arange运算以标量值为起始生成递增/递减索引序列cast本样例基于Reg编程接口实现Cast运算用于RegTensor数据类型转换位宽大转小/小转大compare本样例基于Reg编程接口实现Compare运算用于比较两个向量或向量与标量的大小关系data_relayout本样例基于Reg编程接口演示Interleave和Pack接口实现数据重排Interleave交织 Pack低位提取duplicate本样例基于Reg编程接口实现Duplicate运算标量填充将标量值填充到向量的每个位置gather本样例基于Reg编程接口演示Gather接口源操作数为寄存器按索引收集元素使用LoadAlign/StoreAlign的postUpdate模式histogram本样例基于Reg编程接口实现Histogram运算对输入数据进行直方图统计计算mergemode本样例基于Reg编程接口综合演示MaskMergeMode::MERGING postUpdate UpdateMask ReduceMaxmul本样例基于Reg编程接口实现Mul运算Add/Sub/Div/Max/Min/Prelu接口皆可参考该样例muls本样例基于Reg编程接口实现Muls运算Adds/Maxs/Mins/LeakyRelu接口皆可参考该样例reduce本样例基于Reg编程接口实现Reduce运算SUM模式Reduce接口支持SUM/MAX/MIN归约模式reduce_block本样例基于Reg编程接口实现ReduceDataBlock运算SUM模式对每个DataBlock(32B)内元素归约reduce_pair本样例基于Reg编程接口实现PairReduceElem运算SUM模式对相邻奇偶元素对进行归约求和select本样例基于Reg编程接口实现Select运算根据掩码在两个向量之间选取元素squeeze本样例基于Reg编程接口实现Squeeze运算将向量压缩为标量subnormal本样例基于Reg编程接口实现Div运算演示支持subnormal与不支持subnormal计算结果差异trait本样例基于Reg编程接口实现Div运算使用int64_t类型数据演示单核情况下RegTensor位2倍宽数据处理性能优势truncate本样例基于Reg编程接口实现Truncate运算将浮点数截断到整数位【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考