diff --git a/paddle/phi/kernels/legacy/cpu/one_hot_kernel.cc b/paddle/phi/kernels/legacy/cpu/one_hot_kernel.cc
index d475c5fec98d94..f506b852ce1353 100644
--- a/paddle/phi/kernels/legacy/cpu/one_hot_kernel.cc
+++ b/paddle/phi/kernels/legacy/cpu/one_hot_kernel.cc
@@ -20,17 +20,17 @@
 
 namespace phi {
 
-template <typename DeviceContext, typename InT>
+template <typename Context, typename InT>
 struct OneHotV2OpFunctor {
   const DenseTensor* in_;
   DenseTensor* out_;
   int depth_;
-  const DeviceContext& dev_ctx_;
+  const Context& dev_ctx_;
 
   OneHotV2OpFunctor(const DenseTensor* in,
                     DenseTensor* out,
                     int depth,
-                    const DeviceContext& dev_ctx)
+                    const Context& dev_ctx)
       : in_(in), out_(out), depth_(depth), dev_ctx_(dev_ctx) {}
 
   template <typename OutT>
diff --git a/paddle/phi/kernels/legacy/gpu/one_hot_kernel.cu b/paddle/phi/kernels/legacy/gpu/one_hot_kernel.cu
index 8030231e7fa025..0d331f465a7d9d 100644
--- a/paddle/phi/kernels/legacy/gpu/one_hot_kernel.cu
+++ b/paddle/phi/kernels/legacy/gpu/one_hot_kernel.cu
@@ -40,17 +40,17 @@ __global__ void FillOutputKernel(const InT* p_in_data,
   }
 }
 
-template <typename DeviceContext, typename InT>
+template <typename Context, typename InT>
 struct OneHotV2OpCUDAFunctor {
   const DenseTensor* in_;
   DenseTensor* out_;
-  const DeviceContext& dev_ctx_;
+  const Context& dev_ctx_;
   int depth_;
 
   OneHotV2OpCUDAFunctor(const DenseTensor* in,
                         DenseTensor* out,
                         int depth,
-                        const DeviceContext& dev_ctx)
+                        const Context& dev_ctx)
       : in_(in), out_(out), depth_(depth), dev_ctx_(dev_ctx) {}
 
   template <typename OutT>
diff --git a/paddle/phi/kernels/xpu/reduce.h b/paddle/phi/kernels/xpu/reduce.h
index 65a5f289d483ad..30c544cc9ccb88 100644
--- a/paddle/phi/kernels/xpu/reduce.h
+++ b/paddle/phi/kernels/xpu/reduce.h
@@ -96,8 +96,8 @@ int XPUReduce(const Context& dev_ctx,
   return r;
 }
 
-template <typename DeviceContext, typename T, typename OutT, typename Functor>
-void ReduceKernelImpl(const DeviceContext& dev_ctx,
+template <typename Context, typename T, typename OutT, typename Functor>
+void ReduceKernelImpl(const Context& dev_ctx,
                       const phi::DenseTensor& input,
                       phi::DenseTensor* output,
                       const std::vector<int64_t>& xdims,
@@ -118,8 +118,8 @@ void ReduceKernelImpl(const DeviceContext& dev_ctx,
   }
 }
 
-template <typename DeviceContext, typename T, typename Functor>
-void XPUReduce(const DeviceContext& dev_ctx,
+template <typename Context, typename T, typename Functor>
+void XPUReduce(const Context& dev_ctx,
                const DenseTensor& x,
                const std::vector<int64_t>& dims,
                bool keep_dim,
@@ -142,17 +142,17 @@ void XPUReduce(const DeviceContext& dev_ctx,
     // do reduce sum
     PD_VISIT_XPU_REDUCE_TYPES(
         x.dtype(), "ReduceKernelImpl", ([&] {
-          phi::ReduceKernelImpl<DeviceContext, T, data_t, Functor>(
+          phi::ReduceKernelImpl<Context, T, data_t, Functor>(
               dev_ctx, x, out, xdims, reduce_dims);
         }));
   } else {
     // cast x tensor to out_dtype
-    auto tmp_tensor = phi::Cast<T, DeviceContext>(dev_ctx, x, out_dtype);
+    auto tmp_tensor = phi::Cast<T, Context>(dev_ctx, x, out_dtype);
 
     // do reduce sum
     PD_VISIT_XPU_REDUCE_TYPES(
         out_dtype, "ReduceKernelImpl", ([&] {
-          phi::ReduceKernelImpl<DeviceContext, T, data_t, Functor>(
+          phi::ReduceKernelImpl<Context, T, data_t, Functor>(
               dev_ctx, tmp_tensor, out, xdims, reduce_dims);
         }));