【免费下载链接】atvcATVCAscend C Templates for Vector Compute是为基于Ascend C开发的典型Vector算子封装的一系列模板头文件的集合可帮助用户快速开发典型Vector算子。项目地址: https://gitcode.com/cann/atvcEdge算子样例概述样例概述本样例介绍了利用ATVC实现自定义Edge单算子并完成功能验证算子功能自定义Edge计算的功能(一个元素的结果为周围相邻元素通过自定义计算得到的结果)使用的ATVC模板Pool调用方式Kernel直调样例支持AI处理器型号Ascend 910CAscend 910B算子描述自定义Edge算子数学计算公式输入为二维数组例如 x [ x0, x1, x2, ... x3, x4, x5, ... x6, x7, x8, ...] y4 min(abs(((x2 x5 x8) - (x0 x3 x6)) / 3), 255) 以此类推其他元素的计算结果。自定义Edge算子规格算子类型(OpType)Edge算子输入namewidthheightdata typeformatx10232517floatND算子输出z10232517floatND核函数名EdgeCustom规格限制说明当前模板只支持2维shape按16元素个数对齐、 TILE_LAYOUT{16, 16}、TILE_PADDING{8, 8, 1, 1}的场景目录结构文件名描述edge.cpp自定义Edge算子代码实现以及调用样例算子运行在ascendc-api-adv代码仓目录下执行cd ./ops_templates/atvc/examples bash run_examples.sh edge当前PoolOpTemplate暂不支持ATVC调试调优功能相关功能待后续补充。【免费下载链接】atvcATVCAscend C Templates for Vector Compute是为基于Ascend C开发的典型Vector算子封装的一系列模板头文件的集合可帮助用户快速开发典型Vector算子。项目地址: https://gitcode.com/cann/atvc创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考