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 createAIRRenumberDmaIdPass(); std::unique_ptr createAIRLowerHerdParallelPass(); std::unique_ptr createAIRLabelBroadcastChannelWithTilePass(); std::unique_ptr createAIRCollapseHerdPass(); +std::unique_ptr +createAIRCollapseHerdPass(AIRCollapseHerdPassOptions options); std::unique_ptr createAIRUnrollOuterPerfectlyNestedLoopsPass(); std::unique_ptr 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 createAIRCollapseHerdPass() { return std::make_unique(); } +std::unique_ptr +createAIRCollapseHerdPass(AIRCollapseHerdPassOptions options) { + return std::make_unique(options); +} + std::unique_ptr createAIRUnrollOuterPerfectlyNestedLoopsPass() { return std::make_unique(); }