Skip to content

Fix(BatchMatMulBroadcast3Dx2D)#281

Merged
real-zhangzhe merged 1 commit into
MooreThreads:mainfrom
real-zhangzhe:albert/fix_batch_matmul_3Dx2D
May 26, 2026
Merged

Fix(BatchMatMulBroadcast3Dx2D)#281
real-zhangzhe merged 1 commit into
MooreThreads:mainfrom
real-zhangzhe:albert/fix_batch_matmul_3Dx2D

Conversation

@real-zhangzhe

Copy link
Copy Markdown
Collaborator
  1. 修复了 AFM 模型遇到的 batch matmul 3D x 2D 报错的问题,修复了 mudnn api 参数
  2. 增加了针对这个 case 的单元测试

@real-zhangzhe
real-zhangzhe merged commit 0cf5d12 into MooreThreads:main May 26, 2026
57 of 71 checks passed
@real-zhangzhe
real-zhangzhe deleted the albert/fix_batch_matmul_3Dx2D branch May 27, 2026 05:31
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant