| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
feat: 新增 batch_matmul 批量矩阵乘算子样例（batch 维广播） Co-authored-by: ykxxlzhong<luzhihong3@hisilicon.com> # message auto-generated for no-merge-commit merge: !83 merge bmm into master feat: 新增 batch_matmul 批量矩阵乘算子样例（batch 维广播） Created-by: ykxxlzhong Commit-by: ykxxlzhong Merged-by: cann-robot Description: <!-- 感谢您的合入申请! --> ### 当前PR是否有AI参与: [ ] 否 [x] 是 1. AI Agent 平台: openCode 2. AI 模型: GLM 3. Prompt上下文: 将 cannbot-arena 的 batch_matmul 算子迁移至本仓（样例/测试/CI 注册/文档/注释精简/修复 codecheck 检出问题），人工确认与验收 ### PR功能描述 / 为什么需要这个合入: 新增 batch_matmul 非量化批量矩阵乘算子样例（ samples/matmul/batch_matmul/）： - C[c_batch, M, N] = A[a_batch, M, K] @ B[b_batch, N, K]^T + bias，batch 维按 numpy/torch.matmul 语义广播，支持 rank 2~6，fp16/bf16/fp32（核内 fp32 累加，可选 HF32 快速模式） - 转置由 stride 自动推导（canonical transpose(-1,-2) 视图零拷贝）；bias 经 BT 折入 init MMAD（[N] 共享 / [*c_batch, N] 逐 batch） - host tiling 对齐主线 batch_mat_mul_v3（ResetBaseDav3510 + CalL1TilingDefault + GetBaseK） - 测试：test/matmul/batch_matmul/test_batch_matmul.py，12 个 NPU 用例（fp16/bf16/fp32/HF32、bias、K-major 视图、batch 广播、非对齐尾块/kL1 边界、2-D 输入） - CI 注册：test_config.yaml / operator_list.yaml 新增 matmul/batch_matmul 节点 - 文档：README 算子列表与目录结构、CHANGELOG 条目、性能对比图 figures/batch_matmul.png ### 该PR关联的issue *(格式为fixes #<issue号>, 或者resolves #<issue号>)*: - ### 希望检视人员了解: - 环境为 cannbotdsl 0.6.0，面向 NPU ARCH 3510（Ascend 950PR / Ascend 950DT） - CI 验证命令：bash scripts/ci/run_tests.sh --ops=matmul/batch_matmul --mode=npu See merge request: cann/cannbot-dsl!83 | 1 个月前 | |
add ci Co-authored-by: zhu-lei0614<zhulei113@huawei.com> # message auto-generated for no-merge-commit merge: !24 merge master into master add ci Created-by: zhu-lei0614 Commit-by: zhu-lei0614 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/cannbot-dsl!24 | 2 个月前 |