Star869
1752
代码
代码
Issues219
Pull Requests317
Actions
项目讨论
Wiki
分析
Star869
1752
已合并
bmm支持小m,k轴且n轴=1模板 #6190
zhang-junming21创建于 6月17日
bmm支持小m,k轴且n轴=1模板 #6190
已合并
zhang-junming21创建于 6月17日
从hid54649903/ops-nnbmm_opt_910合入到cann/ops-nn9.1.0
zhang-junming21
zhang-junming21成员
6月17日

描述

通过将计算全部放入vector核优化如下情况性能：
m, k <= 8, n = 1, batch > 100
性能收益：1.4~2.0 (batch轴大小150->40001，且batch轴越大性能优化效果越好）

关联的Issue

Issue #3349

测试

rdv通过，ci流水通过，本地(atk, aclnn测试用例)验证无误
image.png

文档更新

无

类型标签

likedislike
Pull Request已成功合入, 合并人@CANN-robot
（感谢 zhang-junming21 的贡献）
zhang-junming21zhang-junming21成员
6月17日 创建了 pull request，commit 082b9f96
zhang-junming21zhang-junming21成员
6月17日 关联了issue：[Requirement|需求建议]: bmm算子优化m,k较小且n = 1时性能情况
CANN-robotCANN-robot成员
6月17日 添加了label：cann-cla/yes
CANN-robot
CANN-robot成员
6月17日 评论：

Thanks for your pull-request.
The full list of commands accepted by me can be found at here。
You can get sig-info at here

PR Approval Progress

✅ Congratulations! All modules have met the lgtm and approve requirements.

Module Approval Details

module lgtm status approve status
matmul ✅ 唐超, 杨阳 (2/2) ✅ 唐超 (1/1)

💡 Tip:

  • Committer can comment /approve or /lgtm
  • Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

hid54649903, thanks for your pull request. All authors of the commits have signed the CLA. 👍

likedislike
zhang-junming21zhang-junming21成员
6月17日 update merge request[project id: 7665709, iid: 6190, commit_id: 433cbe583de7e87bdd7a12f0be7dc5213c5b6e5b] virtual merging success
zhang-junming21zhang-junming21成员
6月17日 推送  1 个提交:522e44ef-bmm vector feature
zhang-junming21zhang-junming21成员
6月17日 update merge request[project id: 7665709, iid: 6190, commit_id: 61b7dd175434b3078388f8bf9ff78b1e17609203] virtual merging success
CANN-robotCANN-robot成员
6月17日 添加了label：stat/needs-squash
zhang-junming21
zhang-junming21成员
6月17日 评论：

/compile

likedislike
zhang-junming21zhang-junming21成员
6月17日 update merge request[project id: 7665709, iid: 6190, commit_id: 118eb0d28dba0069e339d30b03ed921178624f81] virtual merging success
CANN-robotCANN-robot成员
6月17日 添加了label：ci-pipeline-running
CANN-robot
CANN-robot成员
6月17日 评论：

流水线任务触发成功
任务链接 [d2b5ac24ca3b46868a73ba79b65d94b6][流水线指导]

任务名称状态日志下载链接
Compile_Ascend_X86_mobile_station ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_single ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_experimental ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_950 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM_950 ✅ SUCCESS >>>>> >>>>>
Compile_Pre ✅ SUCCESS >>>>>
pre_comment ✅ SUCCESS >>>>>
Compile_Ascend_X86 ✅ SUCCESS >>>>> >>>>>
Compile_X86_monitor_910b ✅ SUCCESS >>>>> >>>>>
Compile_X86_monitor_910c ✅ SUCCESS >>>>> >>>>>
Compile_X86_monitor_950 ✅ SUCCESS >>>>> >>>>>
UT_Test_ophost ❌ FAILED
UT_Test_opapi ✅ SUCCESS
UT_Test_kernel ✅ SUCCESS
UT_Test_opgraph ✅ SUCCESS
PreSmoke_A900 ✅ SUCCESS >>>>>
PreSmoke_ATK_Test_A2 ✅ SUCCESS >>>>>
likedislike
CANN-robot
CANN-robot成员
6月17日 评论：

流水线任务触发成功
任务链接 [8c9b5b1c86f4412ca86ad1ba93d49e57][流水线指导]

任务名称状态日志下载链接
codecheck ✅ SUCCESS >>>>>
antipoison ✅ SUCCESS >>>>>
codecheck_checkpr ✅ SUCCESS
StaticCheck_codespell_check ✅ SUCCESS
StaticCheck_link_validity_check ✅ SUCCESS
StaticCheck_resource_existence_check ✅ SUCCESS
StaticCheck_tag_closed_check ✅ SUCCESS
StaticCheck_markdownlint ✅ SUCCESS
codecheck_style ✅ SUCCESS >>>>>
codecheck_precommit ⚠️ WARNING >>>>> >>>>>
SCA ✅ SUCCESS >>>>>

[2026-06-17 10:53:05]    CI执行结束

likedislike
CANN-robotCANN-robot成员
6月17日 删除了label：ci-pipeline-running
CANN-robotCANN-robot成员
6月17日 添加了label：ci-pipeline-failed
CANN-robotCANN-robot成员
6月17日 删除了label：ci-pipeline-failed
CANN-robotCANN-robot成员
6月17日 添加了label：ci-pipeline-running
CANN-robot
CANN-robot成员
6月17日 评论：

流水线任务触发成功
任务链接 [f3fb1d9540e2483789591896c4d83a95][流水线指导]

任务名称状态日志下载链接
Compile_Ascend_X86_mobile_station ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_single ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_experimental ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_950 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM_950 ✅ SUCCESS >>>>> >>>>>
Compile_Pre ✅ SUCCESS >>>>>
pre_comment ✅ SUCCESS >>>>>
Compile_Ascend_X86 ✅ SUCCESS >>>>> >>>>>
Compile_X86_monitor_910b ✅ SUCCESS >>>>> >>>>>
Compile_X86_monitor_910c ✅ SUCCESS >>>>> >>>>>
Compile_X86_monitor_950 ✅ SUCCESS >>>>> >>>>>
UT_Test_ophost ❌ FAILED
UT_Test_opapi ✅ SUCCESS
UT_Test_kernel ✅ SUCCESS
UT_Test_opgraph ✅ SUCCESS
PreSmoke_A900 ✅ SUCCESS >>>>>
PreSmoke_ATK_Test_A2 ✅ SUCCESS >>>>>
likedislike
CANN-robotCANN-robot成员
6月17日 删除了label：ci-pipeline-running
CANN-robotCANN-robot成员
6月17日 添加了label：ci-pipeline-failed
CANN-robotCANN-robot成员
6月17日 删除了label：ci-pipeline-failed
CANN-robotCANN-robot成员
6月17日 添加了label：ci-pipeline-running
zhang-junming21
zhang-junming21成员
6月17日 评论：

/compile

likedislike
zhang-junming21zhang-junming21成员
6月17日 update merge request[project id: 7665709, iid: 6190, commit_id: f8db17b1d7d202f5a9fdad41a3b072e88a7376ad] virtual merging success
CANN-robot
CANN-robot成员
6月17日 评论：

流水线任务触发成功
任务链接 [8aad7e64124248849fa4694397a12f5b][流水线指导]

任务名称状态日志下载链接
Compile_Ascend_X86_mobile_station ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_single ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_experimental ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_950 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM_950 ✅ SUCCESS >>>>> >>>>>
Compile_Pre ✅ SUCCESS >>>>>
pre_comment ✅ SUCCESS >>>>>
Compile_Ascend_X86 ✅ SUCCESS >>>>> >>>>>
Compile_X86_monitor_910b ✅ SUCCESS >>>>> >>>>>
Compile_X86_monitor_910c ✅ SUCCESS >>>>> >>>>>
Compile_X86_monitor_950 ✅ SUCCESS >>>>> >>>>>
UT_Test_ophost ✅ SUCCESS
UT_Test_opapi ✅ SUCCESS
UT_Test_kernel ✅ SUCCESS
UT_Test_opgraph ✅ SUCCESS
PreSmoke_A900 ✅ SUCCESS >>>>>
PreSmoke_ATK_Test_A2 ✅ SUCCESS >>>>>

[2026-06-17 11:45:44]    CI执行结束

likedislike
CANN-robot
CANN-robot成员
6月17日 评论：

流水线任务触发成功
任务链接 [19eac7df755e473f92e6a1683defdc61][流水线指导]

任务名称状态日志下载链接
codecheck ✅ SUCCESS >>>>>
antipoison ✅ SUCCESS >>>>>
codecheck_checkpr ✅ SUCCESS
StaticCheck_codespell_check ✅ SUCCESS
StaticCheck_link_validity_check ✅ SUCCESS
StaticCheck_resource_existence_check ✅ SUCCESS
StaticCheck_tag_closed_check ✅ SUCCESS
StaticCheck_markdownlint ✅ SUCCESS
codecheck_style ✅ SUCCESS >>>>>
codecheck_precommit ⚠️ WARNING >>>>> >>>>>
SCA ✅ SUCCESS >>>>>

[2026-06-17 11:40:01]    CI执行结束

likedislike
CANN-robotCANN-robot成员
6月17日 删除了label：ci-pipeline-running
CANN-robotCANN-robot成员
6月17日 添加了label：ci-pipeline-passed
唐
唐超成员
6月17日 评论：

/lgtm
/approve

likedislike
CANN-robotCANN-robot成员
6月17日 添加了label：approved
Y
yangyang016成员
6月17日 评论：

/lgtm

likedislike
CANN-robotCANN-robot成员
6月17日 添加了label：lgtm
CANN-robotCANN-robot成员
6月17日 合入了pull request