CANN/atvc ATVC样例介绍
样例介绍【免费下载链接】atvcATVCAscend C Templates for Vector Compute是为基于Ascend C开发的典型Vector算子封装的一系列模板头文件的集合可帮助用户快速开发典型Vector算子。项目地址: https://gitcode.com/cann/atvc样例名描述模板算子调用方式add使用ATVC的Elementwise模板实现Add算子以及调用样例ElementwiseKernel直调sinh_custom临时Tensor参与计算的自定义Elementwise类算子以及调用样例ElementwiseKernel直调add_with_scalar输入带标量的自定义Elementwise类算子以及调用样例ElementwiseKernel直调reduce_sum使用ATVC的Reduce模板实现自定义ReduceSum算子以及调用样例ReduceKernel直调broadcast_to使用ATVC的Broadcast模板实现自定义BroadcastTo算子以及调用样例BroadcastKernel直调tanh_grad使用Tiling超参进行算子性能调优的ElementWise类算子调用样例ElementwiseKernel直调ops_aclnn使用ATVC基于自定义工程算子的实现以及调用样例单算子API调用ops_pytorch使用ATVC开发自定义算子并实现从PyTorch框架调用的样例PyTorch框架调用add_with_broadcast使用ATVC的Elementwise和Broadcast组合模板实现Add算子以及调用样例BroadcastKernel直调relu_with_reduce_sum使用ATVC的Elementwise和ReduceSum组合模板实现ReduceSumRelu算子以及调用样例ReduceKernel直调edge使用ATVC的Pool模板实现Edge算子以及调用样例PoolKernel直调【免费下载链接】atvcATVCAscend C Templates for Vector Compute是为基于Ascend C开发的典型Vector算子封装的一系列模板头文件的集合可帮助用户快速开发典型Vector算子。项目地址: https://gitcode.com/cann/atvc创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考