[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