| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
feat(custom_op): 新增 TileLang 自定义算子入图/在线编译/离线模型下沉样例 Co-authored-by: qq_45842700<caodazhou@huawei.com> # message auto-generated for no-merge-commit merge: !4520 merge feat/tilelang-add-custom-examples into develop feat(custom_op): 新增 TileLang 自定义算子入图/在线编译/离线模型下沉样例 Created-by: Dazhou_Cao Commit-by: qq_45842700 Merged-by: cann-robot Description: # Pull Request ## 描述 本次 PR 新增三个 TileLang 自定义算子样例，展示如何使用 TileLang 语言开发 Ascend 自定义算子并通过 GE 进行入图、在线编译和离线模型下沉： 1. **tilelang_add_custom** — TileLang 算子入图样例，演示将 TileLang 编译的 kernel 注册为 GE 自定义算子并通过 Session 执行 2. **tilelang_add_custom_online** — TileLang 算子在线编译样例，演示在运行时动态编译 TileLang kernel 并执行 3. **tilelang_add_custom_offline** — TileLang 算子离线 OM 模型下沉样例，演示将 TileLang 算子编译为离线 OM 模型并执行 同时更新了 examples/custom_op/README.md 和 README_en.md，添加 TileLang 样例索引。 ## 变更类型 - [x] ✨ 新功能 ## 关联的Issue ## 如何测试 1. 进入对应样例目录（如 examples/custom_op/tilelang_add_custom） 2. 参考 README.md 配置环境变量（ASCEND_HOME_PATH 等） 3. 执行 bash run.sh 运行样例 4. 验证输出结果与预期一致 ## 核对清单 - [x] 我的代码遵循了项目的代码风格 - [x] 我已对代码进行了自测 - [x] 我已更新了相关的文档 - [x] 我在标题中使用了合适的类型标签（如：feat:, fix:） - [x] 我已经详细阅读了贡献指南（CONTRIBUTING.md），并遵守了其中的所有规定 ## 其他信息 样例参考了同目录下 triton_add_custom 等已有算子样例的结构。每个样例包含完整的源码、CMakeLists.txt、README 和运行脚本。 See merge request: cann/ge!4520 | 1 个月前 |