简介

说明:

本项目提供了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