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