unify pass

Author: gongshaotian

paddle/fluid/pybind/pir.cc

@@ -2582,8 +2582,8 @@ void CheckInferSymbolicIfNeed(Program &program) {  // NOLINT
 void InferSymbolicShapePass(
     std::shared_ptr<pir::PassManager> &pass_manager,  // NOLINT
     pir::Program &program) {                          // NOLINT
-  pir::IrContext *ctx = pir::IrContext::Instance();
-  ctx->GetOrRegisterDialect<pir::shape::ShapeDialect>();
+  // pir::IrContext *ctx = pir::IrContext::Instance();
+  // ctx->GetOrRegisterDialect<pir::shape::ShapeDialect>();
   pass_manager->AddPass(pir::CreateShapeOptimizationPass());
 }
 
@@ -2601,20 +2601,6 @@ std::shared_ptr<Program> ApplyCommonSubexpressionEliminationPass(
   return program;
 }
 
-void ApplyReduceAsToSumPass(
-    std::shared_ptr<pir::PassManager> &pass_manager,  // NOLINT
-    pir::Program &program) {                          // NOLINT
-#ifdef PADDLE_WITH_CINN
-  pass_manager->AddPass(cinn::dialect::ir::CreateReduceAsToSumPass());
-  pass_manager->AddPass(pir::CreateDeadCodeEliminationPass());
-#else
-  PADDLE_THROW(common::errors::Unimplemented(
-      "Currently we only support ReduceAsToSumPass Pass for Pir under "
-      "@to_static, please "
-      "compile PaddlePaddle with CINN"));
-#endif
-}
-
 std::shared_ptr<Program> ApplyFusedBnAddActPass(
     std::shared_ptr<Program> program) {
   pir::PassManager pm(pir::IrContext::Instance(), 3);
@@ -2633,7 +2619,6 @@ void BindIrPass(pybind11::module *m) {
   m->def("infer_symbolic_shape_pass", InferSymbolicShapePass);
   m->def("apply_cse_pass", ApplyCommonSubexpressionEliminationPass);
   m->def("apply_bn_add_act_pass", ApplyFusedBnAddActPass);
-  m->def("reduce_as_sum_pass", ApplyReduceAsToSumPass);
 
   py::class_<Pass, std::shared_ptr<Pass>> pass(*m,
                                                "Pass",

python/paddle/jit/dy2static/pir_partial_program.py

@@ -626,8 +626,9 @@ def apply(self, program):
             # NOTE(gongshaotian): execute infer_symbolic_shape_pass before reduce_as_sum_pass
             pm = paddle.base.libpaddle.pir.PassManager()
             pm.add_pass("delete_assert_op_pass", {})
-            paddle.base.libpaddle.pir.infer_symbolic_shape_pass(pm, program)
-            paddle.base.libpaddle.pir.reduce_as_sum_pass(pm, program)
+            pm.add_pass("shape_optimization_pass", {})
+            pm.add_pass("reduce_as_to_sum_pass", {})
+            pm.add_pass("dead_code_elimination_pass", {})
             pm.run(program)
         return program