[IR] Add control flag for ir inplace pass (#57090)

* add control flag * add log
PaddlePaddle · Sep 8, 2023 · fcb6760 · fcb6760
1 parent 40431e6
commit fcb6760
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 6 deletions.
diff --git a/paddle/fluid/framework/executor_cache.cc b/paddle/fluid/framework/executor_cache.cc
@@ -23,6 +23,9 @@
 #include "paddle/ir/core/value.h"
 #include "paddle/ir/pass/pass.h"
 #include "paddle/ir/pass/pass_manager.h"
+#include "paddle/phi/core/flags.h"
+
+PHI_DECLARE_bool(new_ir_apply_inplace_pass);
 
 namespace paddle {
 namespace framework {
@@ -444,9 +447,11 @@ std::unique_ptr<::ir::Program> ConstructFowardIrProgram(
 
   auto ir_res = paddle::dialect::PdOpLowerToKernelPass(program.get());
 
-  ::ir::PassManager pm(::ir::IrContext::Instance(), 3);
-  pm.AddPass(::ir::CreateInplacePass());
-  pm.Run(ir_res.get());
+  if (FLAGS_new_ir_apply_inplace_pass) {
+    ::ir::PassManager pm(::ir::IrContext::Instance(), 3);
+    pm.AddPass(::ir::CreateInplacePass());
+    pm.Run(ir_res.get());
+  }
 
   return ir_res;
 }
@@ -521,9 +526,11 @@ std::unique_ptr<::ir::Program> ConstructBackwardIrProgram(
 
   auto res = paddle::dialect::PdOpLowerToKernelPass(program.get());
 
-  ::ir::PassManager pm(::ir::IrContext::Instance(), 3);
-  pm.AddPass(::ir::CreateInplacePass());
-  pm.Run(res.get());
+  if (FLAGS_new_ir_apply_inplace_pass) {
+    ::ir::PassManager pm(::ir::IrContext::Instance(), 3);
+    pm.AddPass(::ir::CreateInplacePass());
+    pm.Run(res.get());
+  }
 
   return res;
 }

diff --git a/paddle/fluid/ir/transforms/inplace_pass.cc b/paddle/fluid/ir/transforms/inplace_pass.cc
@@ -330,6 +330,8 @@ class InplacePass : public ir::Pass {
           "is_inplace",
           ir::BoolAttribute::get(ir::IrContext::Instance(), true));
     }
+    LOG_FIRST_N(INFO, 1)
+        << "Apply inplace pass on lowering ::ir::Program to Kernel Dialect.";
   }
 
   bool CanApplyOn(ir::Operation* op) const override {