Replace platform::DeviceContextPool [fluid_ops] (PaddlePaddle#65983)

Author: co63oc

paddle/fluid/framework/ir/auto_mixed_precision_pass.cc

@@ -227,9 +227,9 @@ void AutoMixedPrecisionPass::Init(Graph* graph) const {
         phi::CustomRegisteredDeviceMap::Instance()
             .GetOrRegisterGlobalDeviceTypeId(device_type));
 #else
-    PADDLE_THROW(paddle::platform::errors::Unavailable(
-        "Paddle is not compiled with CustomDevice. "
-        "Cannot enable custom_device_mixed."));
+    PADDLE_THROW(
+        phi::errors::Unavailable("Paddle is not compiled with CustomDevice. "
+                                 "Cannot enable custom_device_mixed."));
 #endif
   }
 

paddle/fluid/framework/ir/conv2d_trans_filter_dilations_nxn_to_1x1_pass.cc

@@ -125,7 +125,7 @@ void Conv2dTransFilterDilationsNxNTo1x1Pass::conv2d_dilation_trans(
         scope->Var(new_weights_name)->GetMutable<phi::DenseTensor>();
     new_weights->Resize({weights_shape[0], weights_shape[1], new_kh, new_kw});
     auto* cpu_ctx = static_cast<phi::CPUContext*>(
-        platform::DeviceContextPool::Instance().Get(phi::CPUPlace()));
+        phi::DeviceContextPool::Instance().Get(phi::CPUPlace()));
     if (weights->dtype() == phi::DataType::FLOAT32) {
       auto weights_data = weights->data<float>();
       auto* new_weights_data = cpu_ctx->Alloc<float>(new_weights);

paddle/fluid/framework/ir/cudnn_placement_pass.cc

@@ -32,7 +32,7 @@ bool CUDNNPlacementPass::IsSupport(const Node* op) const {
     return false;
   }
   for (auto& kernel_pair : it->second) {
-    if (platform::is_gpu_place(kernel_pair.first.place_) &&
+    if (phi::is_gpu_place(kernel_pair.first.place_) &&
         (kernel_pair.first.library_type_ == LibraryType::kCUDNN)) {
       return true;
     }

paddle/fluid/framework/ir/delete_cast_op_pass.cc

@@ -692,7 +692,7 @@ int DeleteCastOpPass::ApplyCastLookupTablePass(ir::Graph* graph) const {
     lookup_table_w->Var()->SetDataType(proto::VarType::FP16);
     if (w_tensor->dtype() != phi::DataType::FLOAT16) {
       auto* cpu_ctx = static_cast<phi::CPUContext*>(
-          platform::DeviceContextPool::Instance().Get(phi::CPUPlace()));
+          phi::DeviceContextPool::Instance().Get(phi::CPUPlace()));
       phi::DenseTensor w_fp32_tensor;
       w_fp32_tensor.Resize(w_tensor->dims());
       w_fp32_tensor.set_type(w_tensor->dtype());

paddle/fluid/framework/ir/delete_cast_op_pass_test.cc

@@ -24,7 +24,7 @@ void AddVarToScope(Scope* param_scope,
   auto* tensor = param_scope->Var(name)->GetMutable<phi::DenseTensor>();
   tensor->Resize(dims);
   auto* cpu_ctx = static_cast<phi::CPUContext*>(
-      platform::DeviceContextPool::Instance().Get(phi::CPUPlace()));
+      phi::DeviceContextPool::Instance().Get(phi::CPUPlace()));
   cpu_ctx->Alloc<float>(tensor);
 }
 

paddle/fluid/framework/ir/delete_quant_dequant_filter_op_pass.cc

@@ -151,7 +151,7 @@ void DeleteQuantDequantFilterOpPass::ApplyImpl(ir::Graph* graph) const {
       const phi::DenseTensor& channel_scale_tensor =
           scope->FindVar(scales_name[0])->Get<phi::DenseTensor>();
       PADDLE_ENFORCE(
-          paddle::platform::is_cpu_place(channel_scale_tensor.place()),
+          phi::is_cpu_place(channel_scale_tensor.place()),
           platform::errors::InvalidArgument(
               "Channel scale tensor's place should be CPU."));
       // compute the channel wise abs max of the weight tensor

paddle/fluid/framework/ir/delete_quant_dequant_linear_op_pass.cc

@@ -139,11 +139,10 @@ void DeleteQuantDequantLinearOpPass::ApplyImpl(ir::Graph* graph) const {
     const phi::DenseTensor& input_scale_tensor =
         scope->GetVar(quantize_linear_op_scale->Name())
             ->Get<phi::DenseTensor>();
-    PADDLE_ENFORCE_EQ(
-        paddle::platform::is_cpu_place(input_scale_tensor.place()),
-        true,
-        platform::errors::InvalidArgument(
-            "Input scale tensor's place should be CPU."));
+    PADDLE_ENFORCE_EQ(phi::is_cpu_place(input_scale_tensor.place()),
+                      true,
+                      platform::errors::InvalidArgument(
+                          "Input scale tensor's place should be CPU."));
 
     float input_scale = NAN;
     if (input_scale_tensor.dtype() == phi::DataType::FLOAT32) {

paddle/fluid/framework/ir/delete_quant_dequant_op_pass.cc

@@ -70,11 +70,10 @@ void DeleteQuantDequantOpPass::ApplyImpl(ir::Graph* graph) const {
             "Scope in DeleteQuantDequantOpPass should not be null."));
     const phi::DenseTensor& input_scale_tensor =
         scope->FindVar(input_scale_var_name)->Get<phi::DenseTensor>();
-    PADDLE_ENFORCE_EQ(
-        paddle::platform::is_cpu_place(input_scale_tensor.place()),
-        true,
-        platform::errors::InvalidArgument(
-            "Input scale tensor's place should be CPU."));
+    PADDLE_ENFORCE_EQ(phi::is_cpu_place(input_scale_tensor.place()),
+                      true,
+                      platform::errors::InvalidArgument(
+                          "Input scale tensor's place should be CPU."));
     const float* input_scale_data = input_scale_tensor.data<float>();
     float input_scale = input_scale_data[0];
 

paddle/fluid/framework/ir/delete_weight_dequant_linear_op_pass_tester.cc

@@ -28,7 +28,7 @@ void AddVarToScope(Scope* param_scope,
   auto* tensor = param_scope->Var(name)->GetMutable<phi::DenseTensor>();
   tensor->Resize(dims);
   auto* dev_ctx = static_cast<phi::CPUContext*>(
-      platform::DeviceContextPool::Instance().Get(phi::CPUPlace()));
+      phi::DeviceContextPool::Instance().Get(phi::CPUPlace()));
   dev_ctx->HostAlloc<T>(tensor, tensor->numel() * sizeof(T));
 }
 

paddle/fluid/framework/ir/fuse_optimizer_ops_pass/fuse_optimizer_op_pass.cc

@@ -293,9 +293,9 @@ bool FuseOptimizerOpPass::OpWithKernelSupportCPUAndGPU(
       kernel_factory.SelectKernelMap(phi::TransToPhiKernelName(op_type));
   bool has_op_kernel = !kernel_key_map.empty() ? true : false;
   for (auto &kernel : kernel_key_map) {
-    if (platform::is_gpu_place(phi::TransToPhiPlace(kernel.first.backend()))) {
+    if (phi::is_gpu_place(phi::TransToPhiPlace(kernel.first.backend()))) {
       support_gpu = true;
-    } else if (platform::is_cpu_place(
+    } else if (phi::is_cpu_place(
                    phi::TransToPhiPlace(kernel.first.backend()))) {
       support_cpu = true;
     }
@@ -308,9 +308,9 @@ bool FuseOptimizerOpPass::OpWithKernelSupportCPUAndGPU(
     if (it != all_kernels.end()) {
       has_op_kernel = true;
       for (auto &kernel_pair : it->second) {
-        if (platform::is_cpu_place(kernel_pair.first.place_)) {
+        if (phi::is_cpu_place(kernel_pair.first.place_)) {
           support_cpu = true;
-        } else if (platform::is_gpu_place(kernel_pair.first.place_)) {
+        } else if (phi::is_gpu_place(kernel_pair.first.place_)) {
           support_gpu = true;
         }
       }