[Mlir-commits] [mlir] [mlir][linalg] Vectorize directly to a named contraction (PR #147296)

Adam Siemieniuk llvmlistbot at llvm.org
Thu Jul 10 03:52:25 PDT 2025


adam-smnk wrote:

Personally, I lean toward forcing users to first decompose unsupported affine maps to not have to guess or compromise on an arbitrary default decomposition for these edge cases.

All the default named op contractions and maps like these:
```mlir
      indexing_maps = [affine_map<(batch, m, n, k) -> (batch, k, m)>,
                       affine_map<(batch, m, n, k) -> (k, n)>,
                       affine_map<(batch, m, n, k) -> (batch, m, n)>]
```
are directly representable by `vector.contract` and should be directly vectorized.


https://github.com/llvm/llvm-project/pull/147296


More information about the Mlir-commits mailing list