CANN ATVC Addcmul算子样例
【免费下载链接】atvcATVCAscend C Templates for Vector Compute是为基于Ascend C开发的典型Vector算子封装的一系列模板头文件的集合可帮助用户快速开发典型Vector算子。项目地址: https://gitcode.com/cann/atvcAddcmul算子样例概述样例概述本样例介绍了利用ATVC实现Addcmul单算子并完成功能验证算子功能Addcmul使用的ATVC模板带前置Elementwise计算和后置Elementwise计算的Broadcast模板调用方式Kernel直调样例支持AI处理器型号Ascend 910CAscend 910B算子描述算子数学计算公式$output_i input_i value * tensor1_i * tensor2_i$算子规格算子类型(OpType)Add算子输入nameshapedata typeformatinput8 * 2048floatNDtensor11 * 2048floatNDtensor21 * 2048floatND算子输出output8 * 2048floatND核函数名AddcmulCustom目录结构文件名描述addcmul.cppAddcmul算子代码实现以及调用样例addcmul.hAddcmul算子代码实现头文件post_compute_add_of_broadcast.h后置Elementwise计算pre_compute_mul_of_broadcast.h)前置Elementwise计算算子运行在ascendc-api-adv代码仓目录下执行cd ./ops_templates/atvc/examples bash run_examples.sh addcmul【免费下载链接】atvcATVCAscend C Templates for Vector Compute是为基于Ascend C开发的典型Vector算子封装的一系列模板头文件的集合可帮助用户快速开发典型Vector算子。项目地址: https://gitcode.com/cann/atvc创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考