AIRSplitL2Memref: Code refactor to improve generality

mlir/include/air/Util/Dependency.h

@@ -73,8 +73,13 @@ bool isAsyncDependent(Operation *a, Operation *b);
 scf::ForOp hoistTargetOpsToNewSCFFor(PatternRewriter &rewriter,
                                      scf::ForOp for_op,
                                      SmallVector<Operation *> target_ops);
-LogicalResult unrollScfParallel(OpBuilder builder, scf::ParallelOp par,
-                                Operation *originalChanOp, IRMapping remap);
+LogicalResult
+unrollScfParallel(OpBuilder builder, scf::ParallelOp par, IRMapping remap,
+                  llvm::DenseMap<Operation *, SmallVector<Operation *>> &opMap);
+LogicalResult unrollScfParallelOnDims(
+    RewriterBase &rewriter, scf::ParallelOp par, IRMapping remap,
+    SmallVector<int> dims,
+    llvm::DenseMap<Operation *, SmallVector<Operation *>> &opMap);
 void populateAIRunrollAIRChannelPutGetInScfParallelPatterns(
     RewritePatternSet &patterns);
 air::WaitAllOp replaceAsyncOpWithWaitAll(OpBuilder builder, IRMapping &remap,

mlir/include/air/Util/Util.h

@@ -271,6 +271,13 @@ void populateBufferMemrefToFuncArgsPattern(RewritePatternSet &patterns);
 Region *findCommonRegionContainingAllAncestors(SmallVector<Operation *> ops,
                                                Operation *until = nullptr);
 
+// A lite version of OperationEquivalence::isRegionEquivalentTo which only
+// checks for const value equivalences.
+bool isRegionEquivalentTo(Region *lhs, Region *rhs);
+// A lite version of OperationEquivalence::isEquivalentTo which only checks for
+// const value equivalences.
+bool isEquivalentTo(Operation *lhs, Operation *rhs);
+
 } // namespace air
 } // namespace xilinx