[MLIR][Arith] add and(a, or(a,b)) folder

[wsmoses]

No description provided.

[llvmbot]

@llvm/pr-subscribers-mlir

@llvm/pr-subscribers-mlir-arith

Author: William Moses (wsmoses)

Changes

---

Full diff: https://github.com/llvm/llvm-project/pull/138998.diff

1 Files Affected:

• (modified) mlir/lib/Dialect/Arith/IR/ArithOps.cpp (+12)

diff --git a/mlir/lib/Dialect/Arith/IR/ArithOps.cpp b/mlir/lib/Dialect/Arith/IR/ArithOps.cpp
index 3b308716c84dc..7cf65cdd4f2da 100644
--- a/mlir/lib/Dialect/Arith/IR/ArithOps.cpp
+++ b/mlir/lib/Dialect/Arith/IR/ArithOps.cpp
@@ -896,6 +896,18 @@ OpFoldResult arith::AndIOp::fold(FoldAdaptor adaptor) {
   if (Value result = foldAndIofAndI(*this))
     return result;
 
+  /// and(a, or(a, b)) -> a
+   for (int i = 0; i < 2; i++) {
+     auto a = getOperand(1 - i);
+     if (auto orOp = getOperand(i).getDefiningOp<arith::OrIOp>()) {
+       for (int j = 0; j < 2; j++) {
+         if (orOp->getOperand(j) == a) {
+           return a;
+         }
+       }
+     }
+   }
+
   return constFoldBinaryOp<IntegerAttr>(
       adaptor.getOperands(),
       [](APInt a, const APInt &b) { return std::move(a) & b; });

[krzysz00]

Question - partly for the peanut gallery - should this be a canonicalization instead of a fold?