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

Cullen Rhodes llvmlistbot at llvm.org
Wed Dec 13 02:54:53 PST 2023


================
@@ -1446,6 +1446,92 @@ 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 > 2 and have unit leading and/or
----------------
c-rhodes wrote:

should be rank > 1
```suggestion
/// for which `a` and `b` are vectors of rank > 1 and have unit leading and/or
```

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


More information about the Mlir-commits mailing list