fix ir default place (PaddlePaddle#57243)

Author: wanghuancoder

paddle/fluid/eager/to_static/run_program_op_node.h

@@ -403,8 +403,13 @@ inline void RunProgramAPI(
 
     if (FLAGS_enable_new_ir_in_executor) {
       // build new ir program
-      auto ir_program = paddle::framework::ConstructFowardIrProgram(
-          forward_global_block, backward_global_block, output_names, x, params);
+      auto ir_program =
+          paddle::framework::ConstructFowardIrProgram(forward_global_block,
+                                                      backward_global_block,
+                                                      output_names,
+                                                      x,
+                                                      params,
+                                                      place);
       interpreter_core =
           paddle::framework::CreateNewIRInterpreterCoreInfoToCache(
               std::move(ir_program),

paddle/fluid/framework/executor_cache.cc

@@ -357,7 +357,8 @@ std::unique_ptr<::pir::Program> ConstructFowardIrProgram(
     const paddle::framework::BlockDesc *backward_global_block,
     const std::vector<std::string> output_names,
     const std::vector<paddle::Tensor> &x,
-    const std::vector<paddle::Tensor> &params) {
+    const std::vector<paddle::Tensor> &params,
+    const phi::Place &place) {
   auto ir_ctx = ::pir::IrContext::Instance();
   auto program = std::make_unique<::pir::Program>(ir_ctx);
 
@@ -381,29 +382,29 @@ std::unique_ptr<::pir::Program> ConstructFowardIrProgram(
     if (block->FindVarRecursive(name) == nullptr) {
       continue;
     }
-    auto place = in_t.place().GetType();
+    auto p = in_t.place().GetType();
 
     auto op_desc = block->PrependOp();
     op_desc->SetType("data");
     op_desc->SetAttr("shape", std::vector<int64_t>());
     // TODO(phlrain) : using tensor dtype
     op_desc->SetAttr("dtype", 0);
-    op_desc->SetAttr("place", static_cast<int>(place));
+    op_desc->SetAttr("place", static_cast<int>(p));
     op_desc->SetAttr("name", name);
     op_desc->SetOutput("out", {name});
   }
 
   std::set<std::string> input_param_names;
   for (auto &param : params) {
     auto &name = param.name();
-    auto place = param.place().GetType();
+    auto p = param.place().GetType();
 
     auto op_desc = local_program.MutableBlock(0)->PrependOp();
     op_desc->SetType("data");
     op_desc->SetAttr("shape", std::vector<int64_t>());
     // TODO(phlrain) : using tensor dtype
     op_desc->SetAttr("dtype", 0);
-    op_desc->SetAttr("place", static_cast<int>(place));
+    op_desc->SetAttr("place", static_cast<int>(p));
     op_desc->SetAttr("name", name);
     op_desc->SetOutput("out", {name});
 
@@ -445,7 +446,7 @@ std::unique_ptr<::pir::Program> ConstructFowardIrProgram(
 
   program_translator.Translate();
 
-  auto ir_res = paddle::dialect::PdOpLowerToKernelPass(program.get());
+  auto ir_res = paddle::dialect::PdOpLowerToKernelPass(program.get(), place);
 
   if (FLAGS_new_ir_apply_inplace_pass) {
     ::pir::PassManager pm(::pir::IrContext::Instance(), 3);

paddle/fluid/framework/executor_cache.h

@@ -254,7 +254,8 @@ std::unique_ptr<::pir::Program> ConstructFowardIrProgram(
     const paddle::framework::BlockDesc* backward_global_block,
     const std::vector<std::string> output_names,
     const std::vector<paddle::Tensor>& x,
-    const std::vector<paddle::Tensor>& params);
+    const std::vector<paddle::Tensor>& params,
+    const phi::Place& place);
 
 std::unique_ptr<::pir::Program> ConstructBackwardIrProgram(
     const paddle::framework::BlockDesc* backward_global_block,