Fixup pass registration to allow for options (Xilinx#493)

nod-ai · Mar 13, 2024 · 7be8043 · 7be8043
1 parent 2e2d2f1
commit 7be8043
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/mlir/include/air/Transform/AIRMiscPasses.h b/mlir/include/air/Transform/AIRMiscPasses.h
@@ -29,6 +29,8 @@ std::unique_ptr<mlir::Pass> createAIRRenumberDmaIdPass();
 std::unique_ptr<mlir::Pass> createAIRLowerHerdParallelPass();
 std::unique_ptr<mlir::Pass> createAIRLabelBroadcastChannelWithTilePass();
 std::unique_ptr<mlir::Pass> createAIRCollapseHerdPass();
+std::unique_ptr<mlir::Pass>
+createAIRCollapseHerdPass(AIRCollapseHerdPassOptions options);
 std::unique_ptr<mlir::Pass> createAIRUnrollOuterPerfectlyNestedLoopsPass();
 std::unique_ptr<mlir::Pass> createAIRUnrollOuterPerfectlyNestedLoopsPass(
     AIRUnrollOuterPerfectlyNestedLoopsPassOptions options);

diff --git a/mlir/lib/Transform/AIRMiscPasses.cpp b/mlir/lib/Transform/AIRMiscPasses.cpp
@@ -1118,6 +1118,11 @@ std::unique_ptr<Pass> createAIRCollapseHerdPass() {
   return std::make_unique<AIRCollapseHerdPass>();
 }
 
+std::unique_ptr<Pass>
+createAIRCollapseHerdPass(AIRCollapseHerdPassOptions options) {
+  return std::make_unique<AIRCollapseHerdPass>(options);
+}
+
 std::unique_ptr<Pass> createAIRUnrollOuterPerfectlyNestedLoopsPass() {
   return std::make_unique<AIRUnrollOuterPerfectlyNestedLoopsPass>();
 }