Star0
0
Code
Code
Issues
Pull Requests
Actions
Discussion
Wiki
Analysis
Star0
0
cann-robotcann-robotfeat: add torch.compile support(take engram_fused_weight as example)
be6e3b2eAdd on 14 days agoHistory
FilesLast commitLast update
adding_an_operator.md
[docs]replace Ascend A5 references with Ascend 950 Co-authored-by: wangliwei12<wangliwei12@huawei.com> # message auto-generated for no-merge-commit merge: !32 merge master into master [docs]replace Ascend A5 references with Ascend 950 Created-by: wangliwei12 Commit-by: wangliwei12 Merged-by: cann-robot Description: <!-- 感谢您的合入申请! --> ### 当前PR是否有AI参与: [x] 否 [ ] 是 __1. AI Agent 平台: __2. AI 模型: __3. Prompt上下文 : ### PR功能描述 / 为什么需要这个合入**: <!-- 本 PR 做了什么，为什么需要 / What does this PR do and why --> ### 该PR关联的issue *(格式为fixes #<issue号>, 或者resolves #<issue号>)*: fixes # ### 希望检视人员了解: ## 改动类型 / Change Type - [ ] Bug 修复 / Bug Fix - [ ] 新功能 / New Feature - [ ] 性能优化 / Performance - [ ] 代码重构 / Refactoring - [ ✅] 文档更新 / Documentation - [ ] 测试相关 / Test - [ ] 其它 / Other ## 测试信息 / Testing <!-- 简要测试说明或关键结果 / Brief test description or key results --> - [ ✅] 单元测试通过 / UT passed - [ ✅] 集成测试通过 / ST passed - [ ✅] 人工验证通过 / Manual verified ## 检查清单 / Checklist - [ ✅] 代码符合规范 / Code follows style guide - [ ✅] 测试添加并通过 / Tests added and passed - [✅ ] 文档已更新 / Docs updated if needed - [ ✅] 无硬编码敏感信息 / No secrets hardcoded - [ ✅] 提交信息符合规范 / Commit message follows convention See merge request: cann/ops-tilelang!32
29 days ago
architecture.md
feat: add torch.compile support(take engram_fused_weight as example) Co-authored-by: ChaoyangJi<jichaoyang@huawei.com> # message auto-generated for no-merge-commit merge: !47 merge master into master feat: add torch.compile support(take engram_fused_weight as example) Created-by: ChaoyangJi Commit-by: ChaoyangJi Merged-by: cann-robot Description: <!-- 感谢您的合入申请! --> ### 当前PR是否有AI参与: [x] 否 [ ] 是 __1. AI Agent 平台: __2. AI 模型: __3. Prompt上下文 : ### PR功能描述 / 为什么需要这个合入**: <!-- 本 PR 做了什么，为什么需要 / What does this PR do and why --> 以engram_fused_weight为例，增加torch.compile的支持，可以支持aclgraph的图接入； 另外，修改： 1. engram_fused_weight通过返回值返回tensor，不通过原来的参数out返回； 2. 增加对pytorch版本的要求 3. 同时支持eager和compile两种方式的调用和测试 ### 该PR关联的issue *(格式为fixes #<issue号>, 或者resolves #<issue号>)*: fixes # ### 希望检视人员了解: ## 改动类型 / Change Type - [ ] Bug 修复 / Bug Fix - [* ] 新功能 / New Feature - [ ] 性能优化 / Performance - [ ] 代码重构 / Refactoring - [ ] 文档更新 / Documentation - [ ] 测试相关 / Test - [ ] 其它 / Other ## 测试信息 / Testing <!-- 简要测试说明或关键结果 / Brief test description or key results --> - [* ] 单元测试通过 / UT passed - [ *] 集成测试通过 / ST passed - [ *] 人工验证通过 / Manual verified ## 检查清单 / Checklist - [ *] 代码符合规范 / Code follows style guide - [ *] 测试添加并通过 / Tests added and passed - [ *] 文档已更新 / Docs updated if needed - [ *] 无硬编码敏感信息 / No secrets hardcoded - [ *] 提交信息符合规范 / Commit message follows convention See merge request: cann/ops-tilelang!47
14 days ago