Add aggressive parallel merging (#262)

* Add aggressive parallel merging

* New infra

* bugfix

* Continue

* functioning

* fix nostore

* Compose rev

Author: wsmoses

include/polygeist/Ops.h

@@ -147,4 +147,84 @@ class BarrierElim final
   }
 };
 
+struct ValueOrInt {
+  bool isValue;
+  mlir::Value v_val;
+  int64_t i_val;
+  ValueOrInt(mlir::Value v) { initValue(v); }
+  void initValue(mlir::Value v) {
+    using namespace mlir;
+    if (v) {
+      IntegerAttr iattr;
+      if (matchPattern(v, m_Constant(&iattr))) {
+        i_val = iattr.getValue().getSExtValue();
+        v_val = nullptr;
+        isValue = false;
+        return;
+      }
+    }
+    isValue = true;
+    v_val = v;
+  }
+
+  ValueOrInt(size_t i) : isValue(false), v_val(), i_val(i) {}
+
+  bool operator>=(int64_t v) {
+    if (isValue)
+      return false;
+    return i_val >= v;
+  }
+  bool operator>(int64_t v) {
+    if (isValue)
+      return false;
+    return i_val > v;
+  }
+  bool operator==(int64_t v) {
+    if (isValue)
+      return false;
+    return i_val == v;
+  }
+  bool operator<(int64_t v) {
+    if (isValue)
+      return false;
+    return i_val < v;
+  }
+  bool operator<=(int64_t v) {
+    if (isValue)
+      return false;
+    return i_val <= v;
+  }
+  bool operator>=(llvm::APInt v) {
+    if (isValue)
+      return false;
+    return i_val >= v.getSExtValue();
+  }
+  bool operator>(llvm::APInt v) {
+    if (isValue)
+      return false;
+    return i_val > v.getSExtValue();
+  }
+  bool operator==(llvm::APInt v) {
+    if (isValue)
+      return false;
+    return i_val == v.getSExtValue();
+  }
+  bool operator<(llvm::APInt v) {
+    if (isValue)
+      return false;
+    return i_val < v.getSExtValue();
+  }
+  bool operator<=(llvm::APInt v) {
+    if (isValue)
+      return false;
+    return i_val <= v.getSExtValue();
+  }
+};
+
+enum class Cmp { EQ, LT, LE, GT, GE };
+
+bool valueCmp(Cmp cmp, mlir::AffineExpr expr, size_t numDim,
+              mlir::ValueRange operands, ValueOrInt val);
+
+bool valueCmp(Cmp cmp, mlir::Value bval, ValueOrInt val);
 #endif // POLYGEISTOPS_H