CANN/ATVC ReluWithReduceSum样例

【免费下载链接】atvcATVCAscend C Templates for Vector Compute是为基于Ascend C开发的典型Vector算子封装的一系列模板头文件的集合可帮助用户快速开发典型Vector算子。项目地址: https://gitcode.com/cann/atvcReluWithReduceSum算子样例概述样例概述本样例介绍了利用ATVC实现ReduceSumElementwise算子并完成功能验证此算子精度适用遵循reduce的精度范围。算子功能ReduceSum算子前置添加Add和后置添加Relu自定义算子使用的ATVC模板带前后Elementwise计算的Reduce模板调用方式Kernel直调样例支持AI处理器型号Ascend 910CAscend 910B算子描述该自定义算子数学计算公式为 $$y \begin{cases} 0 \text{if reduceSum(x 1) 0}\ reduceSum(x 1) \text{if reduceSum(x 1) ≥ 0} \end{cases} $$算子规格算子类型(OpType)ReluWithReduceSum算子输入nameshapedata typeformatx8 * 1024floatND算子输出y1 * 1024floatND核函数名ReluWithReduceSum目录结构文件名描述relu_with_reduce_sum.cpp算子代码实现以及调用样例relu_with_reduce_sum.h算子代码实现头文件pre_compute_add_with_reduce_sum.h前置Elementwise计算post_compute_relu_with_reduce_sum.h后置Elementwise计算算子运行在ascendc-api-adv代码仓目录下执行cd ./ops_templates/atvc/examples bash run_examples.sh relu_with_reduce_sum【免费下载链接】atvcATVCAscend C Templates for Vector Compute是为基于Ascend C开发的典型Vector算子封装的一系列模板头文件的集合可帮助用户快速开发典型Vector算子。项目地址: https://gitcode.com/cann/atvc创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考