From 7be80435096dcc097c2cfff2974f76d037f8c5fa Mon Sep 17 00:00:00 2001
From: erwei-xilinx <erweiw@xilinx.com>
Date: Tue, 12 Mar 2024 18:33:07 -0700
Subject: [PATCH] Fixup pass registration to allow for options (#493)

---
 mlir/include/air/Transform/AIRMiscPasses.h | 2 ++
 mlir/lib/Transform/AIRMiscPasses.cpp       | 5 +++++
 2 files changed, 7 insertions(+)

diff --git a/mlir/include/air/Transform/AIRMiscPasses.h b/mlir/include/air/Transform/AIRMiscPasses.h
index ded7736e0..2da8522f2 100644
--- 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
index 0983300ae..10a97fa4b 100644
--- 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>();
 }