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