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

[Adam Siemieniuk]

adam-smnk wrote:

The impedance mismatch between linalg and vector contractions comes from the fact that `vector.contract` doesn't support broadcast. So, before or during vectorization, the broadcasts have to be decomposed and the extra dimensions created before they go into `vector.contract`.

Today, vectorizer obviously can vectorize such linalg ops with broadcasts.
But I claim the default realization is a bit poor and I wonder if we could do better.


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