已合并
feat: 支持 IndirectLoad Load-Transpose 布局并完善 SIMD/SIMT codegen #2029
xiebangrui2025创建于 23 天前
feat: 支持 IndirectLoad Load-Transpose 布局并完善 SIMD/SIMT codegen #2029
已合并
Pull Request已成功合入, 合并人@CANN-robot
(感谢 xiebangrui2025 的贡献)23 天前 创建了 pull request,commit 20acaddf
atomgit-bot
23 天前 评论:
23 天前 评论:
🤖 正在生成合并请求摘要,请稍候…


不准确?
atomgit-bot
23 天前 评论:
23 天前 评论:
🤖 AI 代码检视正在进行中,请稍候…


不准确?
23 天前 添加了label:stat/needs-squash
23 天前 添加了label:cann-cla/yes
此处折叠了50条消息 查看更多
wangxiaotian995
22 天前 评论:
22 天前 评论:
/lgtm


22 天前 添加了label:lgtm
yangyongqiang
22 天前 评论:
22 天前 评论:
/approve


22 天前 添加了label:approved
22 天前 合入了pull request
Pull Request
描述
本 PR 完整支持 IndirectLoad 输入侧 Load-Transpose 布局,并统一、简化 SIMD/SIMT lowering 与 codegen。主要变更如下:
变更类型
请选择本次引入的变更类型:
关联的Issue
无。
如何测试
描述测试此变更的步骤和前提条件:
cmake --build build --target codegen -j 8,增量编译通过。indirect_load_graph_hint_embedding_slice_e2e_v2、indirect_load_both_transpose_simd_fallback_test_e2e_v2、indirect_load_both_transpose_simd_nddma_test_e2e_v2和indirect_load_both_transpose_simt_test_e2e_v2,4/4 通过。--dry-run --Werror通过;git diff --check通过。scripts/test/run_autofuse_test.sh线上执行列表:双方均为 86 个 IL E2E,双向差集为空。核对清单
其他信息
本 PR 包含 4 个连续 commit:
7d147dec:支持 IndirectLoad Load-Transpose 布局及相关调度协同。41adeac4:统一 lowering metadata,简化 SIMD/SIMT codegen 与地址策略。778a7824:统一 SIMT 输出处理并补充 IL ST。5a1b71fb:修复 dtype/axis/offset 回归并补充 embedding slice E2E。源分支为
xiebangrui2025:IL,目标分支为cann:develop。