diff --git a/include/mlir-hlo/Dialect/mhlo/transforms/passes.h b/include/mlir-hlo/Dialect/mhlo/transforms/passes.h index 269424b..6f662d1 100644 --- a/include/mlir-hlo/Dialect/mhlo/transforms/passes.h +++ b/include/mlir-hlo/Dialect/mhlo/transforms/passes.h @@ -75,8 +75,7 @@ std::unique_ptr createMoveUpDynamicBroadcastsForFusionPass(); /// - Lower rank specialization clusters to SCF and ranked operations. std::unique_ptr createRankSpecializationClusterPass(); std::unique_ptr createRankSpecializationToSCFPass( - int64_t max_target_rank); -std::unique_ptr createRankSpecializationToSCFPass(); + int64_t max_target_rank = 5); std::unique_ptr createOptimizeMhloPass(); std::unique_ptr createLowerComplexPass(); diff --git a/lib/Dialect/mhlo/transforms/rank_specialization.cc b/lib/Dialect/mhlo/transforms/rank_specialization.cc index 60379b1..b00cde4 100644 --- a/lib/Dialect/mhlo/transforms/rank_specialization.cc +++ b/lib/Dialect/mhlo/transforms/rank_specialization.cc @@ -651,8 +651,6 @@ struct LowerRankSpecializationClusterPattern struct RankSpecializationToSCFPass : public RankSpecializationToSCFPassBase { - using RankSpecializationToSCFPassBase< - RankSpecializationToSCFPass>::RankSpecializationToSCFPassBase; explicit RankSpecializationToSCFPass(int64_t max_target_rank) : RankSpecializationToSCFPassBase< RankSpecializationToSCFPass>::RankSpecializationToSCFPassBase() { @@ -699,9 +697,5 @@ std::unique_ptr createRankSpecializationToSCFPass( return std::make_unique(max_target_rank); } -std::unique_ptr createRankSpecializationToSCFPass() { - return std::make_unique(); -} - } // namespace mhlo } // namespace mlir