| 【PR】: torch方式自定义入图
Co-authored-by: wuzhengzhen<wuzhengzhen@huawei.com>
# message auto-generated for no-merge-commit merge:
!867 merge torch_ge_add_custom into develop
【PR】: torch方式自定义入图
Created-by: wuzhengzhen
Commit-by: wuzhengzhen
Merged-by: cann-robot
Description:
# Pull Request
## 描述
请清晰准确地描述本次 Pull Request 的意图和变更内容。
本样例展示了如何使用PyTorch的torch.library机制注册自定义算子,并通过<<<>>>内核调用符调用核函数,以简单的Add算子为例,实现两个向量的逐元素相加。
## 变更类型
请选择本次引入的变更类型:
<!-- [x] 表示选中 -->
- [ ] 🐛 Bug 修复
- [x] ✨ 新功能
- [ ] 💄 代码风格更新(格式化,局部变量)
- [ ] ♻️ 重构(既不修复错误也不增加功能的代码变动)
- [ ] 📦 构建过程或辅助工具的变动
- [ ] 📝 文档内容更新
## 关联的Issue
<!-- 如果这个PR是为了解决特定的Issue,请在当前页面的右侧'关联Issue'部分添加相应Issue链接,并勾选'合并后关闭已关联的 Issue'选项。 -->
## 如何测试
描述测试此变更的步骤和前提条件:
1.在本样例根ascendc_add_custom目录下执行如下步骤,运行该样例。
2. mkdir -p build; cd build
3. cmake ..; make -j
4. python3 ../script/add_custom_test.py
5. 屏幕上若显示:Ran 1 test in **s 则表示用例执行成功。
## 核对清单
<!-- [x] 表示选中 -->
- [x] 我的代码遵循了项目的代码风格
- [x] 我已对代码进行了自测
- [x] 我已更新了相关的文档
- [x] 我在标题中使用了合适的类型标签(如:feat:, fix:)
- [x] 我已经详细阅读了贡献指南(CONTRIBUTING.md),并遵守了其中的所有规定,包括但不限于commit message的格式、无效commit的合并等
## 其他信息
在此添加任何其他关于本次 PR 的说明。
See merge request: cann/ge!867 | 2 个月前 |