已合并
bmm支持小m,k轴且n轴=1模板 #6103
zhang-junming21创建于 6月15日
bmm支持小m,k轴且n轴=1模板 #6103
已合并
zhang-junming21创建于 6月15日
zhang-junming21
zhang-junming21成员
6月15日

描述

通过将计算全部放入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月15日 创建了 pull request,commit a8581ec9
CANN-robotCANN-robot成员
6月15日 添加了label:stat/needs-squash
CANN-robotCANN-robot成员
6月15日 添加了label:cann-cla/yes
CANN-robot
CANN-robot成员
6月15日 评论:

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月15日 修改了pull request 的描述
此处折叠了237条消息 查看更多
CANN-robotCANN-robot成员
6月17日 添加了label:approved
yangyang016成员
6月17日 评论:

/lgtm

likedislike
CANN-robotCANN-robot成员
6月17日 添加了label:lgtm
CANN-robotCANN-robot成员
6月17日 关闭了关联的issue
CANN-robotCANN-robot成员
6月17日 合入了pull request