Delete temp param in eager_gen (#53047)

* Delete temp param in eager_gen

paddle/fluid/eager/api/manual/eager_manual/forwards/add_n_fwd_func.cc

@@ -56,11 +56,9 @@ paddle::Tensor add_n_ad_func(const std::vector<paddle::Tensor>& x) {
           << "add_n_ad_func";
   auto api_result = paddle::experimental::add_n(x);
 
-  std::string forward_trace = "";
   // Check NaN and Inf if needed
   if (FLAGS_check_nan_inf) {
     egr::CheckTensorHasNanOrInf("add_n", api_result);
-    forward_trace = egr::Controller::Instance().GetPythonStack();
   }
 
   // Get Outputs
@@ -89,7 +87,7 @@ paddle::Tensor add_n_ad_func(const std::vector<paddle::Tensor>& x) {
 
     // Set forward's stack
     if (FLAGS_check_nan_inf) {
-      grad_node->SetForwardTrace(forward_trace);
+      grad_node->SetForwardTrace(egr::Controller::Instance().GetPythonStack());
     }
 
     // SetAttributes if needed

paddle/fluid/eager/api/manual/eager_manual/forwards/conv2d_fwd_function.cc

@@ -110,11 +110,9 @@ paddle::Tensor conv2d_ad_func(const paddle::Tensor& input,
                                                  dilations,
                                                  groups,
                                                  data_format);
-  std::string forward_trace = "";
   // Check NaN and Inf if needed
   if (FLAGS_check_nan_inf) {
     egr::CheckTensorHasNanOrInf("conv2d", api_result);
-    forward_trace = egr::Controller::Instance().GetPythonStack();
   }
 
   // Get Outputs
@@ -143,7 +141,7 @@ paddle::Tensor conv2d_ad_func(const paddle::Tensor& input,
 
     // Set forward's stack
     if (FLAGS_check_nan_inf) {
-      grad_node->SetForwardTrace(forward_trace);
+      grad_node->SetForwardTrace(egr::Controller::Instance().GetPythonStack());
     }
 
     // SetAttributes if needed

paddle/fluid/eager/api/manual/eager_manual/forwards/sync_batch_norm_fwd_func.cc

@@ -172,11 +172,9 @@ sync_batch_norm__ad_func(const paddle::Tensor& x,
                                              data_layout,
                                              use_global_stats,
                                              trainable_statistics);
-  std::string forward_trace = "";
   // Check NaN and Inf if needed
   if (FLAGS_check_nan_inf) {
     egr::CheckTensorHasNanOrInf("sync_batch_norm_", api_result);
-    forward_trace = egr::Controller::Instance().GetPythonStack();
   }
 
   // Get Outputs
@@ -231,7 +229,7 @@ sync_batch_norm__ad_func(const paddle::Tensor& x,
 
     // Set forward's stack
     if (FLAGS_check_nan_inf) {
-      grad_node->SetForwardTrace(forward_trace);
+      grad_node->SetForwardTrace(egr::Controller::Instance().GetPythonStack());
     }
 
     egr::Controller::Instance().PushBackForceSequentialNodes(grad_node.get());

paddle/fluid/eager/auto_code_generator/generator/eager_gen.py

@@ -299,7 +299,7 @@ FORWARD_BODY_TEMPLATE = """  if(require_any_grad) {{
 {}
     // Set for forward trace
   if (FLAGS_check_nan_inf) {{
-  {}
+    grad_node->SetForwardTrace(egr::Controller::Instance().GetPythonStack());
   }}
     // SetAttributes if needed
 {}
@@ -490,10 +490,8 @@ CHECK_BACKWARD_INPLACE_TEMPLATE = """
   }}"""
 
 CHECK_NAN_AND_INF_TEMPLATE_FORWARD = """
-  std::string forward_trace ="";
   if (FLAGS_check_nan_inf) {{
       egr::CheckTensorHasNanOrInf("{}", {});
-      forward_trace = egr::Controller::Instance().GetPythonStack();
   }}
 """
 
@@ -1070,15 +1068,11 @@ class DygraphFunctionGeneratorBase(FunctionGeneratorBase):
 
         node_event_name = forward_api_name + " node_creation"
         node_creation_event_str = f"{indent}paddle::platform::RecordEvent node_creation_record_event(\"{node_event_name}\", paddle::platform::TracerEventType::OperatorInner, 1);\n"
-        set_forward_trace = (
-            f"{indent} grad_node->SetForwardTrace(forward_trace);"
-        )
         if not for_backward:
             self.node_creation_str = FORWARD_BODY_TEMPLATE.format(
                 node_creation_event_str,
                 pass_stop_gradient_args_str,
                 node_construction_str,
-                set_forward_trace,
                 set_attributes_str,
                 set_input_tensor_wrappers_str,
                 set_grad_out_meta_str,

paddle/fluid/pybind/eager_utils.cc

@@ -218,6 +218,7 @@ std::shared_ptr<imperative::VarBase> CastPyArg2VarBase(PyObject* obj,
 
 void SetPythonStack() {
   if (FLAGS_check_nan_inf) {
+    VLOG(4) << "this is SetPythonStack";
     pybind11::gil_scoped_acquire gil;
     PyObject* mod = PyImport_ImportModule("traceback");
     PyObject* traceback_list = PyObject_CallMethod(mod, "format_stack", "");