PaddlePaddle
diff --git a/‎paddle/fluid/eager/amp_utils.h
Copy file name to clipboard
-165Lines changed: 0 additions & 165 deletions b/‎paddle/fluid/eager/amp_utils.h
Copy file name to clipboard
-165Lines changed: 0 additions & 165 deletions
diff --git a/‎paddle/fluid/eager/api/manual/eager_manual/forwards/add_n_fwd_func.cc
Copy file name to clipboard
+5-4Lines changed: 5 additions & 4 deletions b/‎paddle/fluid/eager/api/manual/eager_manual/forwards/add_n_fwd_func.cc
Copy file name to clipboard
+5-4Lines changed: 5 additions & 4 deletions
diff --git a/‎paddle/fluid/eager/api/manual/eager_manual/forwards/conv2d_fwd_function.cc
Copy file name to clipboard
+6-6Lines changed: 6 additions & 6 deletions b/‎paddle/fluid/eager/api/manual/eager_manual/forwards/conv2d_fwd_function.cc
Copy file name to clipboard
+6-6Lines changed: 6 additions & 6 deletions
diff --git a/‎paddle/fluid/eager/api/manual/eager_manual/forwards/multiply_fwd_func.cc
Copy file name to clipboard
+13-8Lines changed: 13 additions & 8 deletions b/‎paddle/fluid/eager/api/manual/eager_manual/forwards/multiply_fwd_func.cc
Copy file name to clipboard
+13-8Lines changed: 13 additions & 8 deletions
diff --git a/‎paddle/fluid/eager/api/manual/eager_manual/forwards/sync_batch_norm_fwd_func.cc
Copy file name to clipboard
-2Lines changed: 0 additions & 2 deletions b/‎paddle/fluid/eager/api/manual/eager_manual/forwards/sync_batch_norm_fwd_func.cc
Copy file name to clipboard
-2Lines changed: 0 additions & 2 deletions
diff --git a/‎paddle/fluid/eager/api/manual/fluid_manual/forwards/fused_attention_fwd_func.cc
Copy file name to clipboard
+3-3Lines changed: 3 additions & 3 deletions b/‎paddle/fluid/eager/api/manual/fluid_manual/forwards/fused_attention_fwd_func.cc
Copy file name to clipboard
+3-3Lines changed: 3 additions & 3 deletions
diff --git a/‎paddle/fluid/eager/api/manual/fluid_manual/forwards/fused_bias_dropout_residual_layer_norm_fwd_func.cc
Copy file name to clipboard
+2-2Lines changed: 2 additions & 2 deletions b/‎paddle/fluid/eager/api/manual/fluid_manual/forwards/fused_bias_dropout_residual_layer_norm_fwd_func.cc
Copy file name to clipboard
+2-2Lines changed: 2 additions & 2 deletions
diff --git a/‎paddle/fluid/eager/api/manual/fluid_manual/forwards/fused_feedforward_fwd_func.cc
Copy file name to clipboard
+3-3Lines changed: 3 additions & 3 deletions b/‎paddle/fluid/eager/api/manual/fluid_manual/forwards/fused_feedforward_fwd_func.cc
Copy file name to clipboard
+3-3Lines changed: 3 additions & 3 deletions
diff --git a/‎paddle/fluid/eager/api/manual/fluid_manual/forwards/fused_gate_attention_fwd_func.cc
Copy file name to clipboard
+3-3Lines changed: 3 additions & 3 deletions b/‎paddle/fluid/eager/api/manual/fluid_manual/forwards/fused_gate_attention_fwd_func.cc
Copy file name to clipboard
+3-3Lines changed: 3 additions & 3 deletions
@@ -13,12 +13,11 @@
 // limitations under the License.
 
 #include "paddle/common/flags.h"
-#include "paddle/fluid/eager/amp_utils.h"
 #include "paddle/fluid/eager/api/manual/eager_manual/dygraph_forward_api.h"
 #include "paddle/fluid/eager/api/manual/eager_manual/nodes/nodes.h"
 #include "paddle/fluid/eager/api/utils/global_utils.h"
-#include "paddle/fluid/eager/eager_amp_auto_cast.h"
 #include "paddle/fluid/eager/nan_inf_utils.h"
+#include "paddle/fluid/imperative/amp_utils.h"
 #include "paddle/fluid/platform/profiler/event_tracing.h"
 
 COMMON_DECLARE_bool(check_nan_inf);
@@ -36,9 +35,11 @@ paddle::Tensor add_n_ad_func(const std::vector<paddle::Tensor>& x) {
     paddle::small_vector<std::vector<paddle::Tensor>, egr::kSlotSmallVectorSize>
         amp_tensors_vector = {x};
 
-    auto amp_dst_dtype = egr::GetAmpDestDtype(op_name, amp_tensors_vector);
+    auto amp_dst_dtype =
+        paddle::imperative::GetAmpDestDtype(op_name, amp_tensors_vector);
 
-    auto NEW_x = egr::EagerAmpAutoCasts("x", x, amp_dst_dtype, op_name);
+    auto NEW_x =
+        paddle::imperative::AmpAutoCasts("x", x, amp_dst_dtype, op_name);
 
     {
       paddle::imperative::AutoCastGuard guard(
 
@@ -13,13 +13,12 @@
 // limitations under the License.
 
 #include "paddle/common/flags.h"
-#include "paddle/fluid/eager/amp_utils.h"
 #include "paddle/fluid/eager/api/manual/eager_manual/dygraph_forward_api.h"
 #include "paddle/fluid/eager/api/manual/eager_manual/nodes/nodes.h"
 #include "paddle/fluid/eager/api/utils/global_utils.h"
-#include "paddle/fluid/eager/eager_amp_auto_cast.h"
 #include "paddle/fluid/eager/eager_layout_auto_tune.h"
 #include "paddle/fluid/eager/nan_inf_utils.h"
+#include "paddle/fluid/imperative/amp_utils.h"
 #include "paddle/fluid/platform/profiler/event_tracing.h"
 
 COMMON_DECLARE_bool(check_nan_inf);
@@ -44,12 +43,13 @@ paddle::Tensor conv2d_ad_func(const paddle::Tensor& input,
     paddle::small_vector<std::vector<paddle::Tensor>, egr::kSlotSmallVectorSize>
         amp_tensors_vector = {{input}, {filter}};
 
-    auto amp_dst_dtype = egr::GetAmpDestDtype(op_name, amp_tensors_vector);
+    auto amp_dst_dtype =
+        paddle::imperative::GetAmpDestDtype(op_name, amp_tensors_vector);
 
     auto new_input =
-        egr::EagerAmpAutoCast("input", input, amp_dst_dtype, op_name);
-    auto new_filter =
-        egr::EagerAmpAutoCast("filter", filter, amp_dst_dtype, op_name);
+        paddle::imperative::AmpAutoCast("input", input, amp_dst_dtype, op_name);
+    auto new_filter = paddle::imperative::AmpAutoCast(
+        "filter", filter, amp_dst_dtype, op_name);
 
     {
       paddle::imperative::AutoCastGuard guard(
 
@@ -13,14 +13,13 @@
 // limitations under the License.
 
 #include "paddle/common/flags.h"
-#include "paddle/fluid/eager/amp_utils.h"
 #include "paddle/fluid/eager/api/manual/eager_manual/dygraph_forward_api.h"
 #include "paddle/fluid/eager/api/manual/eager_manual/nodes/nodes.h"
 #include "paddle/fluid/eager/api/utils/global_utils.h"
-#include "paddle/fluid/eager/eager_amp_auto_cast.h"
 #include "paddle/fluid/eager/eager_layout_auto_tune.h"
 #include "paddle/fluid/eager/nan_inf_utils.h"
 #include "paddle/fluid/eager/type_promotion_utils.h"
+#include "paddle/fluid/imperative/amp_utils.h"
 #include "paddle/fluid/platform/place.h"
 #include "paddle/fluid/platform/profiler/event_tracing.h"
 #include "paddle/phi/api/include/sparse_api.h"
@@ -45,10 +44,13 @@ paddle::Tensor multiply_ad_func(const paddle::Tensor& x,
     paddle::small_vector<std::vector<paddle::Tensor>, egr::kSlotSmallVectorSize>
         amp_tensors_vector = {{x}, {y}};
 
-    auto amp_dst_dtype = egr::GetAmpDestDtype(op_name, amp_tensors_vector);
+    auto amp_dst_dtype =
+        paddle::imperative::GetAmpDestDtype(op_name, amp_tensors_vector);
 
-    auto new_x = egr::EagerAmpAutoCast("x", x, amp_dst_dtype, op_name);
-    auto new_y = egr::EagerAmpAutoCast("y", y, amp_dst_dtype, op_name);
+    auto new_x =
+        paddle::imperative::AmpAutoCast("x", x, amp_dst_dtype, op_name);
+    auto new_y =
+        paddle::imperative::AmpAutoCast("y", y, amp_dst_dtype, op_name);
 
     {
       paddle::imperative::AutoCastGuard guard(
@@ -392,10 +394,13 @@ paddle::Tensor multiply_ad_func(const paddle::Tensor& x,
     paddle::small_vector<std::vector<paddle::Tensor>, egr::kSlotSmallVectorSize>
         amp_tensors_vector = {{x}, {y}};
 
-    auto amp_dst_dtype = egr::GetAmpDestDtype(op_name, amp_tensors_vector);
+    auto amp_dst_dtype =
+        paddle::imperative::GetAmpDestDtype(op_name, amp_tensors_vector);
 
-    auto new_x = egr::EagerAmpAutoCast("x", x, amp_dst_dtype, op_name);
-    auto new_y = egr::EagerAmpAutoCast("y", y, amp_dst_dtype, op_name);
+    auto new_x =
+        paddle::imperative::AmpAutoCast("x", x, amp_dst_dtype, op_name);
+    auto new_y =
+        paddle::imperative::AmpAutoCast("y", y, amp_dst_dtype, op_name);
 
     {
       paddle::imperative::AutoCastGuard guard(
 
@@ -13,11 +13,9 @@
 // limitations under the License.
 
 #include "paddle/common/flags.h"
-#include "paddle/fluid/eager/amp_utils.h"
 #include "paddle/fluid/eager/api/manual/eager_manual/dygraph_forward_api.h"
 #include "paddle/fluid/eager/api/manual/eager_manual/nodes/nodes.h"
 #include "paddle/fluid/eager/api/utils/global_utils.h"
-#include "paddle/fluid/eager/eager_amp_auto_cast.h"
 #include "paddle/fluid/eager/eager_layout_auto_tune.h"
 #include "paddle/fluid/eager/nan_inf_utils.h"
 #include "paddle/fluid/platform/profiler/event_tracing.h"
 
@@ -14,10 +14,10 @@
 
 #include "paddle/fluid/eager/accumulation/accumulation_node.h"
 #include "paddle/fluid/eager/amp_auto_cast.h"
-#include "paddle/fluid/eager/amp_utils.h"
 #include "paddle/fluid/eager/api/manual/fluid_manual/dygraph_forward_api.h"
 #include "paddle/fluid/eager/api/manual/fluid_manual/nodes/nodes.h"
 #include "paddle/fluid/eager/api/utils/global_utils.h"
+#include "paddle/fluid/imperative/amp_utils.h"
 #include "paddle/fluid/platform/profiler/event_tracing.h"
 
 std::tuple<paddle::Tensor,
@@ -76,8 +76,8 @@ fused_attention_dygraph_function(
     if (Ln2Scale.initialized()) amp_tensors_vector.push_back({Ln2Scale});
     if (Ln2Bias.initialized()) amp_tensors_vector.push_back({Ln2Bias});
 
-    auto amp_dst_dtype =
-        egr::GetAmpDestDtype("fused_attention", amp_tensors_vector);
+    auto amp_dst_dtype = paddle::imperative::GetAmpDestDtype(
+        "fused_attention", amp_tensors_vector);
 
     auto NEW_X = egr::AmpAutoCast("X", X, amp_dst_dtype, "fused_attention");
     auto NEW_QKVW =
 
@@ -14,10 +14,10 @@
 
 #include "paddle/fluid/eager/accumulation/accumulation_node.h"
 #include "paddle/fluid/eager/amp_auto_cast.h"
-#include "paddle/fluid/eager/amp_utils.h"
 #include "paddle/fluid/eager/api/manual/fluid_manual/dygraph_forward_api.h"
 #include "paddle/fluid/eager/api/manual/fluid_manual/nodes/nodes.h"
 #include "paddle/fluid/eager/api/utils/global_utils.h"
+#include "paddle/fluid/imperative/amp_utils.h"
 #include "paddle/fluid/platform/profiler/event_tracing.h"
 
 std::tuple<paddle::Tensor,
@@ -49,7 +49,7 @@ fused_bias_dropout_residual_layer_norm_dygraph_function(
     if (LnScale.initialized()) amp_tensors_vector.push_back({LnScale});
     if (LnBias.initialized()) amp_tensors_vector.push_back({LnBias});
 
-    auto amp_dst_dtype = egr::GetAmpDestDtype(
+    auto amp_dst_dtype = paddle::imperative::GetAmpDestDtype(
         "fused_bias_dropout_residual_layer_norm", amp_tensors_vector);
 
     auto NEW_X = egr::AmpAutoCast(
 
@@ -13,10 +13,10 @@
 // limitations under the License.
 
 #include "paddle/fluid/eager/amp_auto_cast.h"
-#include "paddle/fluid/eager/amp_utils.h"
 #include "paddle/fluid/eager/api/manual/fluid_manual/dygraph_forward_api.h"
 #include "paddle/fluid/eager/api/manual/fluid_manual/nodes/nodes.h"
 #include "paddle/fluid/eager/api/utils/global_utils.h"
+#include "paddle/fluid/imperative/amp_utils.h"
 #include "paddle/fluid/platform/profiler/event_tracing.h"
 
 std::tuple<paddle::Tensor,
@@ -67,8 +67,8 @@ fused_feedforward_dygraph_function(
     if (Ln2Scale.initialized()) amp_tensors_vector.push_back({Ln2Scale});
     if (Ln2Bias.initialized()) amp_tensors_vector.push_back({Ln2Bias});
 
-    auto amp_dst_dtype =
-        egr::GetAmpDestDtype("fused_feedforward", amp_tensors_vector);
+    auto amp_dst_dtype = paddle::imperative::GetAmpDestDtype(
+        "fused_feedforward", amp_tensors_vector);
 
     auto NEW_X = egr::AmpAutoCast("X", X, amp_dst_dtype, "fused_feedforward");
     auto NEW_Linear1Weight = egr::AmpAutoCast(
 
@@ -13,10 +13,10 @@
 // limitations under the License.
 
 #include "paddle/fluid/eager/amp_auto_cast.h"
-#include "paddle/fluid/eager/amp_utils.h"
 #include "paddle/fluid/eager/api/manual/fluid_manual/dygraph_forward_api.h"
 #include "paddle/fluid/eager/api/manual/fluid_manual/nodes/nodes.h"
 #include "paddle/fluid/eager/api/utils/global_utils.h"
+#include "paddle/fluid/imperative/amp_utils.h"
 #include "paddle/fluid/platform/profiler/event_tracing.h"
 
 std::tuple<paddle::Tensor,
@@ -66,8 +66,8 @@ fused_gate_attention_dygraph_function(
     if (GateWeight.initialized()) amp_tensors_vector.push_back({GateWeight});
     if (GateBias.initialized()) amp_tensors_vector.push_back({GateBias});
 
-    auto amp_dst_dtype =
-        egr::GetAmpDestDtype("fused_gate_attention", amp_tensors_vector);
+    auto amp_dst_dtype = paddle::imperative::GetAmpDestDtype(
+        "fused_gate_attention", amp_tensors_vector);
 
     auto NEW_Query =
         egr::AmpAutoCast("Query", Query, amp_dst_dtype, "fused_gate_attention");