[Mlir-commits] [mlir] [mlir] [linalg] fix failure on specializing matmul with permuted loops (PR #184294)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Tue Mar 3 04:27:12 PST 2026
ziereis wrote:
I would have to check for the "test for broadcasts", im not sure about this. However i'm not sure i fully understand how this would look like.
Is this something you were thinking about?
A: (b,m,n,k) -> (m,k) //broadcasts along batch dim
B: (b,m,n,k) -> (b, k,n)
C: (b,m,n,k) -> (b, m,n)
Would this be something that should be specialized to a batch_matmul or should this be left as a generic?
https://github.com/llvm/llvm-project/pull/184294
More information about the Mlir-commits
mailing list