简介
说明:
- 关于AI Core的介绍请参见《Ascend C算子开发》中“概念原理和术语 > 硬件架构与数据处理原理”。
本项目提供了AI Core算子的开发和调用样例,请开发者根据实际情况参考对应实现。
目录说明
├── example
│ ├── add_example # AddExample算子的目录名,一般小写下划线形式
│ │ ├── CMakeLists.txt # 算子编译配置文件,保留原文件即可
│ │ ├── examples # 算子使用示例
│ │ ├── op_graph # 算子构图相关目录
│ │ ├── op_host # 算子信息库、Tiling、InferShape相关实现
│ │ ├── op_kernel # 算子kernel目录
│ │ └── op_kernel_aicpu # 算子kernel_aicpu目录
│ ├── mc2 # 通算融合类算子示例
│ │ ├── all_gather_add # AllGatherAdd算子的目录名,一般小写下划线形式
│ │ │ └── ...
│ ├── CMakeLists.txt # 算子编译配置文件,保留原文件即可
│ └── README.md # 算子说明文档
算子开发样例
| 样例目录 | 样例介绍 | 算子开发 | 算子调用 |
|---|---|---|---|
| add_example | 实现两个张量相加功能的算子。 | 算子端到端开发过程参见AI Core算子开发指南。 | 调用样例参见README |
| mc2/all_gather_add | 先进行 AllGather 集合通信,再执行逐元素相加。 | 算子端到端开发过程参见AI Core算子开发指南。 | 调用样例参见README |