Replace framework::LoDTensorArray part1 (#66912)

co63oc · web-flow · commit da335106dddf · 2024-08-02T10:50:07.000+08:00
diff --git a/paddle/fluid/eager/api/manual/eager_manual/forwards/add_n_fwd_func.cc b/paddle/fluid/eager/api/manual/eager_manual/forwards/add_n_fwd_func.cc
@@ -24,7 +24,7 @@ COMMON_DECLARE_bool(check_nan_inf);
 
 paddle::Tensor add_n_ad_func(const std::vector<paddle::Tensor>& x) {
   // Dygraph Record Event
-  paddle::platform::RecordEvent dygraph_entrance_record_event(
+  phi::RecordEvent dygraph_entrance_record_event(
       "add_n dygraph", paddle::platform::TracerEventType::Operator, 1);
 
   // AMP Logic
@@ -76,7 +76,7 @@ paddle::Tensor add_n_ad_func(const std::vector<paddle::Tensor>& x) {
 
   // Node Creation
   if (require_any_grad) {
-    paddle::platform::RecordEvent node_creation_record_event(
+    phi::RecordEvent node_creation_record_event(
         "add_n node_creation",
         paddle::platform::TracerEventType::OperatorInner,
         1);
diff --git a/paddle/fluid/eager/api/manual/eager_manual/forwards/conv2d_fwd_function.cc b/paddle/fluid/eager/api/manual/eager_manual/forwards/conv2d_fwd_function.cc
@@ -32,7 +32,7 @@ paddle::Tensor conv2d_ad_func(const paddle::Tensor& input,
                               int groups,
                               std::string data_format) {
   // Dygraph Record Event
-  paddle::platform::RecordEvent dygraph_entrance_record_event(
+  phi::RecordEvent dygraph_entrance_record_event(
       "conv2d dygraph", paddle::platform::TracerEventType::Operator, 1);
 
   // AMP Logic
@@ -129,7 +129,7 @@ paddle::Tensor conv2d_ad_func(const paddle::Tensor& input,
 
   // Node Creation
   if (require_any_grad) {
-    paddle::platform::RecordEvent node_creation_record_event(
+    phi::RecordEvent node_creation_record_event(
         "conv2d node_creation",
         paddle::platform::TracerEventType::OperatorInner,
         1);
diff --git a/paddle/fluid/eager/api/manual/eager_manual/forwards/multiply_fwd_func.cc b/paddle/fluid/eager/api/manual/eager_manual/forwards/multiply_fwd_func.cc
@@ -42,7 +42,7 @@ paddle::Tensor multiply_ad_func(const paddle::Tensor& x,
   VLOG(3) << "Running AD API: "
           << "multiply";
   // Dygraph Record Event
-  paddle::platform::RecordEvent dygraph_entrance_record_event(
+  phi::RecordEvent dygraph_entrance_record_event(
       "multiply dygraph", paddle::platform::TracerEventType::Operator, 1);
 
   // AMP Logic
@@ -153,7 +153,7 @@ paddle::Tensor multiply_ad_func(const paddle::Tensor& x,
 
   // Node Creation
   if (require_any_grad) {
-    paddle::platform::RecordEvent node_creation_record_event(
+    phi::RecordEvent node_creation_record_event(
         "multiply node_creation",
         paddle::platform::TracerEventType::OperatorInner,
         1);
@@ -240,7 +240,7 @@ paddle::Tensor& multiply__ad_func(paddle::Tensor& x,  // NOLINT
   VLOG(3) << "Running AD API: "
           << "multiply_";
   // Dygraph Record Event
-  paddle::platform::RecordEvent dygraph_entrance_record_event(
+  phi::RecordEvent dygraph_entrance_record_event(
       "multiply_ dygraph", paddle::platform::TracerEventType::Operator, 1);
 
   // AMP Logic
@@ -319,7 +319,7 @@ paddle::Tensor& multiply__ad_func(paddle::Tensor& x,  // NOLINT
   std::shared_ptr<MultiplyGradNode> grad_node;
   // Set grad_node before API Call
   if (require_any_grad) {
-    paddle::platform::RecordEvent node_creation_record_event(
+    phi::RecordEvent node_creation_record_event(
         "multiply node_creation",
         paddle::platform::TracerEventType::OperatorInner,
         1);
@@ -412,7 +412,7 @@ paddle::Tensor multiply_ad_func(const paddle::Tensor& x,
   VLOG(3) << "Running AD API: "
           << "multiply";
   // Dygraph Record Event
-  paddle::platform::RecordEvent dygraph_entrance_record_event(
+  phi::RecordEvent dygraph_entrance_record_event(
       "multiply dygraph", paddle::platform::TracerEventType::Operator, 1);
 
   // AMP Logic
@@ -522,7 +522,7 @@ paddle::Tensor multiply_ad_func(const paddle::Tensor& x,
 
   // Node Creation
   if (require_any_grad) {
-    paddle::platform::RecordEvent node_creation_record_event(
+    phi::RecordEvent node_creation_record_event(
         "multiply node_creation",
         paddle::platform::TracerEventType::OperatorInner,
         1);
diff --git a/paddle/fluid/eager/api/manual/eager_manual/forwards/reshard_fwd_func.cc b/paddle/fluid/eager/api/manual/eager_manual/forwards/reshard_fwd_func.cc
@@ -25,7 +25,7 @@ paddle::Tensor reshard_ad_function(
   VLOG(3) << "Running AD API: "
           << "reshard dygraph";
   // Dygraph Record Event
-  paddle::platform::RecordEvent dygraph_entrance_record_event(
+  phi::RecordEvent dygraph_entrance_record_event(
       "reshard dygraph", paddle::platform::TracerEventType::Communication, 1);
 
   // Get Input AutoGradMeta
@@ -40,7 +40,7 @@ paddle::Tensor reshard_ad_function(
 
   // Set grad_node before API Call
   if (require_any_grad) {
-    paddle::platform::RecordEvent node_creation_record_event(
+    phi::RecordEvent node_creation_record_event(
         "reshard node_creation",
         paddle::platform::TracerEventType::Communication,
         1);
diff --git a/paddle/fluid/eager/api/manual/eager_manual/forwards/sync_batch_norm_fwd_func.cc b/paddle/fluid/eager/api/manual/eager_manual/forwards/sync_batch_norm_fwd_func.cc
@@ -46,7 +46,7 @@ sync_batch_norm__ad_func(const paddle::Tensor& x,
   VLOG(3) << "Running AD API: "
           << "sync_batch_norm_";
   // Dygraph Record Event
-  paddle::platform::RecordEvent dygraph_entrance_record_event(
+  phi::RecordEvent dygraph_entrance_record_event(
       "sync_batch_norm_ dygraph",
       paddle::platform::TracerEventType::Operator,
       1);
@@ -209,7 +209,7 @@ sync_batch_norm__ad_func(const paddle::Tensor& x,
 
   // Node Creation
   if (require_any_grad) {
-    paddle::platform::RecordEvent node_creation_record_event(
+    phi::RecordEvent node_creation_record_event(
         "sync_batch_norm_ node_creation",
         paddle::platform::TracerEventType::OperatorInner,
         1);
@@ -388,7 +388,7 @@ sync_batch_norm__ad_func(const paddle::Tensor& x,
   VLOG(3) << "Running AD API: "
           << "sync_batch_norm_";
   // Dygraph Record Event
-  paddle::platform::RecordEvent dygraph_entrance_record_event(
+  phi::RecordEvent dygraph_entrance_record_event(
       "sync_batch_norm_ dygraph",
       paddle::platform::TracerEventType::Operator,
       1);
@@ -551,7 +551,7 @@ sync_batch_norm__ad_func(const paddle::Tensor& x,
 
   // Node Creation
   if (require_any_grad) {
-    paddle::platform::RecordEvent node_creation_record_event(
+    phi::RecordEvent node_creation_record_event(
         "sync_batch_norm_ node_creation",
         paddle::platform::TracerEventType::OperatorInner,
         1);
diff --git a/paddle/fluid/eager/api/manual/eager_manual/nodes/add_n_node.cc b/paddle/fluid/eager/api/manual/eager_manual/nodes/add_n_node.cc
@@ -43,7 +43,7 @@ AddNGradNodeFinal::operator()(
   //    accumulation when the output(s) of corresponding forward OP are shared
   //    by other OP(s), which may have extra accumulation overhead than
   //    'Local_XXXGradNode'.
-  paddle::platform::RecordEvent node_execution_inner(
+  phi::RecordEvent node_execution_inner(
       "Local_AddNGradNodeFinal",
       paddle::platform::TracerEventType::OperatorInner,
       1);
diff --git a/paddle/fluid/eager/api/manual/eager_manual/nodes/conv2d_nodes.cc b/paddle/fluid/eager/api/manual/eager_manual/nodes/conv2d_nodes.cc
@@ -46,7 +46,7 @@ Conv2dGradNodeFinal::operator()(
   //    accumulation when the output(s) of corresponding forward OP are shared
   //    by other OP(s), which may have extra accumulation overhead than
   //    'Local_XXXGradNode'.
-  paddle::platform::RecordEvent node_execution_inner(
+  phi::RecordEvent node_execution_inner(
       "Local_Conv2dGradNodeFinal",
       paddle::platform::TracerEventType::OperatorInner,
       1);
@@ -130,7 +130,7 @@ Conv2dGradNodeFinal::operator()(
 
   // Create Grad Node
   if (trace_backward) {
-    paddle::platform::RecordEvent node_creation_record_event(
+    phi::RecordEvent node_creation_record_event(
         "conv2d_grad node_creation",
         paddle::platform::TracerEventType::OperatorInner,
         1);
@@ -229,7 +229,7 @@ Conv2dDoubleGradNodeFinal::operator()(
   //    accumulation when the output(s) of corresponding forward OP are shared
   //    by other OP(s), which may have extra accumulation overhead than
   //    'Local_XXXGradNode'.
-  paddle::platform::RecordEvent node_execution_inner(
+  phi::RecordEvent node_execution_inner(
       "Local_Conv2dDoubleGradNodeFinal",
       paddle::platform::TracerEventType::OperatorInner,
       1);
diff --git a/paddle/fluid/eager/api/manual/eager_manual/nodes/multiply_node.cc b/paddle/fluid/eager/api/manual/eager_manual/nodes/multiply_node.cc
@@ -49,7 +49,7 @@ MultiplyGradNode::operator()(
   //    accumulation when the output(s) of corresponding forward OP are shared
   //    by other OP(s), which may have extra accumulation overhead than
   //    'Local_XXXGradNode'.
-  paddle::platform::RecordEvent node_execution_inner(
+  phi::RecordEvent node_execution_inner(
       "Local_MultiplyGradNode",
       paddle::platform::TracerEventType::OperatorInner,
       1);
@@ -175,7 +175,7 @@ MultiplyGradNode::operator()(
 
   if (!paddle::prim::PrimCommonUtils::IsEagerPrimEnabled() || need_skip) {
     if (trace_backward) {
-      paddle::platform::RecordEvent node_creation_record_event(
+      phi::RecordEvent node_creation_record_event(
           "multiply_grad node_creation",
           paddle::platform::TracerEventType::OperatorInner,
           1);
@@ -266,7 +266,7 @@ MultiplyDoubleGradNode::operator()(
   //    accumulation when the output(s) of corresponding forward OP are shared
   //    by other OP(s), which may have extra accumulation overhead than
   //    'Local_XXXGradNode'.
-  paddle::platform::RecordEvent node_execution_inner(
+  phi::RecordEvent node_execution_inner(
       "Local_MultiplyDoubleGradNode",
       paddle::platform::TracerEventType::OperatorInner,
       1);
@@ -539,7 +539,7 @@ MultiplyGradNode::operator()(
   //    accumulation when the output(s) of corresponding forward OP are shared
   //    by other OP(s), which may have extra accumulation overhead than
   //    'Local_XXXGradNode'.
-  paddle::platform::RecordEvent node_execution_inner(
+  phi::RecordEvent node_execution_inner(
       "Local_MultiplyGradNode",
       paddle::platform::TracerEventType::OperatorInner,
       1);
diff --git a/paddle/fluid/eager/api/manual/eager_manual/nodes/reshard_node.cc b/paddle/fluid/eager/api/manual/eager_manual/nodes/reshard_node.cc
@@ -38,7 +38,7 @@ ReshardGradNode::operator()(
   //    accumulation when the output(s) of corresponding forward OP are shared
   //    by other OP(s), which may have extra accumulation overhead than
   //    'Local_XXXGradNode'.
-  paddle::platform::RecordEvent node_execution_inner(
+  phi::RecordEvent node_execution_inner(
       "Local_ReshardGradNode",
       paddle::platform::TracerEventType::OperatorInner,
       1);
diff --git a/paddle/fluid/eager/api/manual/eager_manual/nodes/sync_batch_norm_node.cc b/paddle/fluid/eager/api/manual/eager_manual/nodes/sync_batch_norm_node.cc
@@ -46,7 +46,7 @@ SyncBatchNormGradNode::operator()(
   //    accumulation when the output(s) of corresponding forward OP are shared
   //    by other OP(s), which may have extra accumulation overhead than
   //    'Local_XXXGradNode'.
-  paddle::platform::RecordEvent node_execution_inner(
+  phi::RecordEvent node_execution_inner(
       "Local_SyncBatchNormGradNode",
       paddle::platform::TracerEventType::OperatorInner,
       1);
@@ -278,7 +278,7 @@ SyncBatchNormGradNode::operator()(
   //    accumulation when the output(s) of corresponding forward OP are shared
   //    by other OP(s), which may have extra accumulation overhead than
   //    'Local_XXXGradNode'.
-  paddle::platform::RecordEvent node_execution_inner(
+  phi::RecordEvent node_execution_inner(
       "Local_SyncBatchNormGradNode",
       paddle::platform::TracerEventType::OperatorInner,
       1);
diff --git a/paddle/fluid/eager/api/manual/fluid_manual/forwards/fused_attention_fwd_func.cc b/paddle/fluid/eager/api/manual/fluid_manual/forwards/fused_attention_fwd_func.cc
@@ -53,7 +53,7 @@ fused_attention_dygraph_function(
     const paddle::Tensor& Ln2Scale,
     const paddle::Tensor& Ln2Bias,
     const paddle::framework::AttributeMap& attr_map) {
-  paddle::platform::RecordEvent dygraph_entrance_record_event(
+  phi::RecordEvent dygraph_entrance_record_event(
       "fused_attention dygraph",
       paddle::platform::TracerEventType::Operator,
       1);
@@ -323,7 +323,7 @@ fused_attention_dygraph_function(
   egr::EagerUtils::GetOutput(outs["Y"][0], &Y);
 
   {
-    paddle::platform::RecordEvent node_creation_record_event(
+    phi::RecordEvent node_creation_record_event(
         "fused_attention node_creation",
         paddle::platform::TracerEventType::Operator,
         1);
diff --git a/paddle/fluid/eager/api/manual/fluid_manual/forwards/fused_bias_dropout_residual_layer_norm_fwd_func.cc b/paddle/fluid/eager/api/manual/fluid_manual/forwards/fused_bias_dropout_residual_layer_norm_fwd_func.cc
@@ -32,7 +32,7 @@ fused_bias_dropout_residual_layer_norm_dygraph_function(
     const paddle::Tensor& LnScale,
     const paddle::Tensor& LnBias,
     const paddle::framework::AttributeMap& attr_map) {
-  paddle::platform::RecordEvent dygraph_entrance_record_event(
+  phi::RecordEvent dygraph_entrance_record_event(
       "fused_bias_dropout_residual_layer_norm dygraph",
       paddle::platform::TracerEventType::Operator,
       1);
@@ -162,7 +162,7 @@ fused_bias_dropout_residual_layer_norm_dygraph_function(
   egr::EagerUtils::GetOutput(outs["Y"][0], &Y);
 
   {
-    paddle::platform::RecordEvent node_creation_record_event(
+    phi::RecordEvent node_creation_record_event(
         "fused_bias_dropout_residual_layer_norm node_creation",
         paddle::platform::TracerEventType::OperatorInner,
         1);
diff --git a/paddle/fluid/eager/api/manual/fluid_manual/forwards/fused_feedforward_fwd_func.cc b/paddle/fluid/eager/api/manual/fluid_manual/forwards/fused_feedforward_fwd_func.cc
@@ -43,7 +43,7 @@ fused_feedforward_dygraph_function(
     const paddle::Tensor& Ln2Scale,
     const paddle::Tensor& Ln2Bias,
     const paddle::framework::AttributeMap& attr_map) {
-  paddle::platform::RecordEvent dygraph_entrance_record_event(
+  phi::RecordEvent dygraph_entrance_record_event(
       "fused_feedforward dygraph",
       paddle::platform::TracerEventType::Operator,
       1);
@@ -270,7 +270,7 @@ fused_feedforward_dygraph_function(
   egr::EagerUtils::GetOutput(outs["Dropout2Out"][0], &Dropout2Out);
 
   {
-    paddle::platform::RecordEvent node_creation_record_event(
+    phi::RecordEvent node_creation_record_event(
         "fused_feedforward node_creation",
         paddle::platform::TracerEventType::Operator,
         1);
diff --git a/paddle/fluid/eager/api/manual/fluid_manual/forwards/fused_gate_attention_fwd_func.cc b/paddle/fluid/eager/api/manual/fluid_manual/forwards/fused_gate_attention_fwd_func.cc
@@ -42,7 +42,7 @@ fused_gate_attention_dygraph_function(
     const paddle::Tensor& OutLinearWeight,
     const paddle::Tensor& OutLinearBias,
     const paddle::framework::AttributeMap& attr_map) {
-  paddle::platform::RecordEvent dygraph_entrance_record_event(
+  phi::RecordEvent dygraph_entrance_record_event(
       "fused_gate_attention dygraph",
       paddle::platform::TracerEventType::Operator,
       1);
@@ -270,7 +270,7 @@ fused_gate_attention_dygraph_function(
   egr::EagerUtils::GetOutput(outs["Out"][0], &Out);
 
   {
-    paddle::platform::RecordEvent node_creation_record_event(
+    phi::RecordEvent node_creation_record_event(
         "fused_gate_attention node_creation",
         paddle::platform::TracerEventType::Operator,
         1);
diff --git a/paddle/fluid/eager/api/manual/fluid_manual/forwards/fused_gemm_epilogue_fwd_func.cc b/paddle/fluid/eager/api/manual/fluid_manual/forwards/fused_gemm_epilogue_fwd_func.cc
@@ -25,7 +25,7 @@ paddle::Tensor fused_gemm_epilogue_dygraph_function(
     const paddle::Tensor& Y,
     const paddle::Tensor& Bias,
     const paddle::framework::AttributeMap& attr_map) {
-  paddle::platform::RecordEvent dygraph_entrance_record_event(
+  phi::RecordEvent dygraph_entrance_record_event(
       "fused_gemm_epilogue dygraph",
       paddle::platform::TracerEventType::Operator,
       1);
@@ -93,7 +93,7 @@ paddle::Tensor fused_gemm_epilogue_dygraph_function(
   egr::EagerUtils::GetOutput(outs["Out"][0], &Out);
 
   {
-    paddle::platform::RecordEvent node_creation_record_event(
+    phi::RecordEvent node_creation_record_event(
         "fused_gemm_epilogue node_creation",
         paddle::platform::TracerEventType::OperatorInner,
         1);
diff --git a/paddle/fluid/eager/auto_code_generator/eager_generator.cc b/paddle/fluid/eager/auto_code_generator/eager_generator.cc
@@ -1372,7 +1372,7 @@ static std::string GenerateGradNodeCreationContent(
       "%s\n"
       "%s"
       "  {\n"
-      "    paddle::platform::RecordEvent node_creation_record_event(\"%s\", "
+      "    phi::RecordEvent node_creation_record_event(\"%s\", "
       "paddle::platform::TracerEventType::OperatorInner, 1);\n"
       "%s"
       "    if(require_any_grad) {\n"
@@ -2096,7 +2096,7 @@ static std::pair<std::string, std::string> GenerateForwardFunctionContents(
   }
 
   const char* DYGRAPH_FUNCTION_EVENT_RECORD_FUNCTION_TEMPLATE =
-      "  paddle::platform::RecordEvent dygraph_entrance_record_event(\"%s\", "
+      "  phi::RecordEvent dygraph_entrance_record_event(\"%s\", "
       "paddle::platform::TracerEventType::Operator, 1);";
   std::string event_name = op_type + " dygraph";
   std::string fwd_record_event_str = paddle::string::Sprintf(
diff --git a/paddle/fluid/eager/auto_code_generator/generator/eager_gen.py b/paddle/fluid/eager/auto_code_generator/generator/eager_gen.py
@@ -277,7 +277,7 @@ class {} : public egr::GradNodeBase {{
    // * 'Local_XXXGradNode' will only cover execution time of this function.
    // * 'Global_XXXGradNode' will not only cover execution time of this function, but also include gradient
    //    accumulation when the output(s) of corresponding forward OP are shared by other OP(s), which may have extra accumulation overhead than 'Local_XXXGradNode'.
-  paddle::platform::RecordEvent grad_node_record_event_inner(\"Local_{}\", paddle::platform::TracerEventType::OperatorInner, 1);
+  phi::RecordEvent grad_node_record_event_inner(\"Local_{}\", paddle::platform::TracerEventType::OperatorInner, 1);
 
   // Fill Zero For GradIn Tensors
 {}
@@ -1308,7 +1308,7 @@ def GenerateNodeCreationCodes(self, for_backward=False, is_inplaced=False):
         set_grad_in_meta_str = "\n".join(set_grad_in_meta_list)
 
         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"
+        node_creation_event_str = f"{indent}phi::RecordEvent node_creation_record_event(\"{node_event_name}\", paddle::platform::TracerEventType::OperatorInner, 1);\n"
         self.node_creation_str = ""
         if not for_backward:
             self.node_creation_before_call_str = (
@@ -1887,7 +1887,7 @@ def GenerateForwardDefinitionAndDeclaration(self, is_inplaced):
                 inputs_call_args_str_tmp = ", ".join(self.inputs_call_list_tmp)
                 forward_call_str = f"{indent}{api_out_type} api_result = paddle::experimental::{namespace}{function_name}({inputs_call_args_str_tmp});"
 
-        dygraph_event_str = f"{indent}paddle::platform::RecordEvent dygraph_entrance_record_event(\"{forward_api_name} dygraph\", paddle::platform::TracerEventType::Operator, 1);\n"
+        dygraph_event_str = f"{indent}phi::RecordEvent dygraph_entrance_record_event(\"{forward_api_name} dygraph\", paddle::platform::TracerEventType::Operator, 1);\n"
         log_memory_info_str = f"{indent}paddle::memory::LogDeviceMemoryStats(egr::Controller::Instance().GetExpectedPlace(), \"{forward_api_name}\");"
         forward_ad_function_name = GetDygraphForwardFunctionName(
             forward_api_name
diff --git a/paddle/fluid/eager/auto_code_generator/generator/python_c_gen.py b/paddle/fluid/eager/auto_code_generator/generator/python_c_gen.py
@@ -100,7 +100,7 @@ def FindParsingFunctionFromAttributeType(atype):
 """
 
 
-RECORD_EVENT_TEMPLATE = "paddle::platform::RecordEvent {}(\"{} {}\", paddle::platform::TracerEventType::UserDefined, 1);"
+RECORD_EVENT_TEMPLATE = "phi::RecordEvent {}(\"{} {}\", paddle::platform::TracerEventType::UserDefined, 1);"
 
 
 RETURN_INPLACE_PYOBJECT_TEMPLATE = """
diff --git a/paddle/fluid/eager/backward.cc b/paddle/fluid/eager/backward.cc
diff --git a/paddle/fluid/eager/custom_operator/custom_operator_node.cc b/paddle/fluid/eager/custom_operator/custom_operator_node.cc
diff --git a/paddle/fluid/eager/to_static/run_program_op_node.h b/paddle/fluid/eager/to_static/run_program_op_node.h
