[Mlir-commits] [mlir] [mlir][vector] Add pattern to drop unit dim from elementwise(a, b)) (PR #74817)

[Andrzej Warzyński]

================
@@ -1446,6 +1446,89 @@ struct ChainedReduction final : OpRewritePattern<vector::ReductionOp> {
   }
 };
 
+/// Replace:
+///   elementwise(a, b)
+/// with:
+///   sc_a = shape_cast(a)
+///   sc_b = shape_cast(b)
+///   res = elementwise(sc_a, sc_b)
+///   return shape_cast(res)
+/// for which `a` and `b` are vectors of rank > 1 and have unit leading and/or
+/// trailing dimension.
----------------
banach-space wrote:

This is good feedback and I am happy to incorporate it before landing this, thanks Ben!

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