已合并
aclnnTransposeBatchMatMul支持B、K非对齐，PermX2[0,2,1]，B*K>65535等场景 #4240
ww-blue创建于 4月25日
aclnnTransposeBatchMatMul支持B、K非对齐，PermX2[0,2,1]，B*K>65535等场景 #4240
已合并
Pull Request已成功合入, 合并人@CANN-robot
（感谢 ww-blue 的贡献）
CANN-robot
4月25日 评论：
4月25日 评论：
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 |
|---|---|---|
| */*/docs/acl*.md | ✅ 吴瑶, 杨露露 (2/2) | ✅ 吴瑶 (1/1) |
| matmul | ✅ 唐超, 胡碧霞 (2/2) | ✅ 唐超, 胡碧霞 (2/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
ww-blue, thanks for your pull request. All authors of the commits have signed the CLA. 👍
4月25日 添加了label：cann-cla/yes
CANN-robot4月25日 将sxb154714,chaotang233,crystalhu,yangyang016,fanqirui,zhou-qilong,chenqi317,LinPX,liubo75,tangweiwei2,wangzitao_leo,zhang-wu设为评审人
4月25日 将sxb154714,chaotang233,crystalhu,yangyang016,fanqirui,zhou-qilong,chenqi317,LinPX,liubo75,tangweiwei2,wangzitao_leo,zhang-wu设为评审人
CANN-robot4月25日 将sxb154714,chaotang233,crystalhu,yangyang016,fanqirui,chenqi317,LinPX,liubo75,tangweiwei2,wangzitao_leo,zhang-wu设为审查人
4月25日 将sxb154714,chaotang233,crystalhu,yangyang016,fanqirui,chenqi317,LinPX,liubo75,tangweiwei2,wangzitao_leo,zhang-wu设为审查人
此处折叠了133条消息 查看更多
4月29日 添加了label：lgtmapproved
4月29日 关闭了关联的issue
4月29日 合入了pull request
描述
aclnnTransposeBatchMatMul支持B、K非对齐，PermX2(0,2,1)，B×K>65535等场景
具体来说：
在Atlas A2/A3平台，更改如下限制：
放开(M, B, K)输入时的B * K <= 65535限制
当x1的输入shape为(B, M, K)时，K <= 65535；当x1的输入shape为(M, B, K)时，B * K <= 65535
↓
当x1的输入shape为(B, M, K)时，需要K <= 65535; 当x1的输入shape为(M, B, K)且B * K > 65535时，不支持传入scale，并且batchSplitFactor只能等于1, permX1必须为[1, 0, 2]。
放开B/K对齐限制：
删除：x2的第二维或x2的第三维不能被16整除。
支持PermX2=[0,2,1]
permX2仅支持输入[0, 1, 2]。
↓
当permX2输入为[0, 2, 1]时，不支持传入scale，并且batchSplitFactor只能等于1, permX1必须为[1, 0, 2]。
关联的Issue
关联Issue #2416
测试
文档更新
ops-nn/matmul/transpose_batch_mat_mul/docs/aclnnTransposeBatchMatMul.md
类型标签