Rename ctx to dev_ctx in paddle/phi/kernels/sparse/ directory (#73163)

co63oc · web-flow · commit ffb7f5f55192 · 2025-06-09T18:03:22.000+08:00
diff --git a/paddle/phi/kernels/selected_rows/hsigmoid_loss_grad_kernel.cc b/paddle/phi/kernels/selected_rows/hsigmoid_loss_grad_kernel.cc
@@ -37,7 +37,7 @@ static std::vector<int64_t> PathToRows(const DenseTensor& path) {
 }
 
 template <typename T, typename Context>
-void HSigmoidLossGradKernel(const Context& ctx,
+void HSigmoidLossGradKernel(const Context& dev_ctx,
                             const DenseTensor& x,
                             const DenseTensor& w,
                             const DenseTensor& label,
@@ -62,7 +62,7 @@ void HSigmoidLossGradKernel(const Context& ctx,
   phi::DDim temp_dim(w.dims());
   temp_dim[0] = static_cast<int>(real_rows.size());
   w_grad_value->Resize(temp_dim);
-  phi::HSigmoidLossGradKernelImpl<T>(ctx,
+  phi::HSigmoidLossGradKernelImpl<T>(dev_ctx,
                                      x,
                                      w,
                                      label,
diff --git a/paddle/phi/kernels/selected_rows/hsigmoid_loss_grad_kernel.h b/paddle/phi/kernels/selected_rows/hsigmoid_loss_grad_kernel.h
@@ -21,7 +21,7 @@ namespace phi {
 namespace sr {
 
 template <typename T, typename Context>
-void HSigmoidLossGradKernel(const Context& ctx,
+void HSigmoidLossGradKernel(const Context& dev_ctx,
                             const DenseTensor& x,
                             const DenseTensor& w,
                             const DenseTensor& label,
diff --git a/paddle/phi/kernels/selected_rows/impl/isfinite_kernel_impl.h b/paddle/phi/kernels/selected_rows/impl/isfinite_kernel_impl.h
@@ -19,16 +19,16 @@
 
 namespace phi {
 
-#define DEFINE_ISFINITE_SR(isfinite)                                  \
-  template <typename T, typename Context>                             \
-  void isfinite##SR(                                                  \
-      const Context& ctx, const SelectedRows& x, SelectedRows* out) { \
-    if (out && out->numel() == 0) {                                   \
-      ctx.template Alloc<bool>(out);                                  \
-      return;                                                         \
-    }                                                                 \
-    ctx.template Alloc<bool>(out);                                    \
-    Isinf##Kernel<T, Context>(ctx, x.value(), out->mutable_value());  \
+#define DEFINE_ISFINITE_SR(isfinite)                                      \
+  template <typename T, typename Context>                                 \
+  void isfinite##SR(                                                      \
+      const Context& dev_ctx, const SelectedRows& x, SelectedRows* out) { \
+    if (out && out->numel() == 0) {                                       \
+      dev_ctx.template Alloc<bool>(out);                                  \
+      return;                                                             \
+    }                                                                     \
+    dev_ctx.template Alloc<bool>(out);                                    \
+    Isinf##Kernel<T, Context>(dev_ctx, x.value(), out->mutable_value());  \
   }
 
 DEFINE_ISFINITE_SR(Isinf)
diff --git a/paddle/phi/kernels/selected_rows/impl/lamb_kernel_impl.h b/paddle/phi/kernels/selected_rows/impl/lamb_kernel_impl.h
@@ -174,7 +174,7 @@ void ComputeRowImpl(const Context& dev_ctx,
   DenseTensor trust_ratio_div;
   trust_ratio_div.Resize(param.dims());
   /*auto trust_ratio_div =
-      ctx.AllocateTmpTensor<MT, DeviceContext>(param.dims(), dev_ctx);*/
+      dev_ctx.AllocateTmpTensor<MT, DeviceContext>(param.dims(), dev_ctx);*/
   auto* trust_ratio_div_ptr = dev_ctx.template Alloc<MT>(&trust_ratio_div);
 
   const void* param_ptr = param.data();
diff --git a/paddle/phi/kernels/selected_rows/isfinite_kernel.h b/paddle/phi/kernels/selected_rows/isfinite_kernel.h
@@ -22,7 +22,7 @@ namespace phi {
 #define DEFINE_ISFINITE_SR(isfinite_sr)   \
   template <typename T, typename Context> \
   void isfinite_sr(                       \
-      const Context& ctx, const SelectedRows& x, SelectedRows* out);
+      const Context& dev_ctx, const SelectedRows& x, SelectedRows* out);
 
 DEFINE_ISFINITE_SR(IsinfSR)
 DEFINE_ISFINITE_SR(IsnanSR)
diff --git a/paddle/phi/kernels/selected_rows/shape_kernel.cc b/paddle/phi/kernels/selected_rows/shape_kernel.cc
@@ -24,17 +24,17 @@ limitations under the License. */
 namespace phi::sr {
 
 template <typename T, typename Context>
-void ShapeKernel(const Context& ctx,
+void ShapeKernel(const Context& dev_ctx,
                  const SelectedRows& input,
                  DenseTensor* out) {
-  phi::ShapeKernel<T, Context>(ctx, input.value(), out);
+  phi::ShapeKernel<T, Context>(dev_ctx, input.value(), out);
 }
 
 template <typename T, typename Context>
-void Shape64Kernel(const Context& ctx,
+void Shape64Kernel(const Context& dev_ctx,
                    const SelectedRows& input,
                    DenseTensor* out) {
-  phi::Shape64Kernel<T, Context>(ctx, input.value(), out);
+  phi::Shape64Kernel<T, Context>(dev_ctx, input.value(), out);
 }
 
 }  // namespace phi::sr
diff --git a/paddle/phi/kernels/selected_rows/shape_kernel.h b/paddle/phi/kernels/selected_rows/shape_kernel.h
@@ -20,12 +20,12 @@ namespace phi {
 namespace sr {
 
 template <typename T, typename Context>
-void ShapeKernel(const Context& ctx,
+void ShapeKernel(const Context& dev_ctx,
                  const SelectedRows& input,
                  DenseTensor* out);
 
 template <typename T, typename Context>
-void Shape64Kernel(const Context& ctx,
+void Shape64Kernel(const Context& dev_ctx,
                    const SelectedRows& input,
                    DenseTensor* out);
 
diff --git a/paddle/phi/kernels/sparse/gpu/gather_gemm_scatter.h b/paddle/phi/kernels/sparse/gpu/gather_gemm_scatter.h
@@ -34,7 +34,7 @@ template <int ComputeCapability,
           typename Output,
           typename IntT>
 void GatherGemmScatterDriver(
-    const phi::GPUContext& ctx,
+    const phi::GPUContext& dev_ctx,
     const size_t key,
     const Input* const a,
     const Input* const b,
@@ -63,7 +63,7 @@ void GatherGemmScatterDriver(
                                       in_type,                                \
                                       out_type,                               \
                                       int32_t>(                               \
-      const phi::GPUContext& ctx,                                             \
+      const phi::GPUContext& dev_ctx,                                         \
       const size_t key,                                                       \
       const in_type* const a,                                                 \
       const in_type* const b,                                                 \
@@ -82,11 +82,11 @@ void GatherGemmScatterDriver(
         autotune::MakeGatherGemmScatterTuner<transpose_a, transpose_b>(       \
             kernels[0]);                                                      \
     for (auto i = 1; i < kernels.size(); i++) tuner->AddCallBack(kernels[i]); \
-    tuner->Run(ctx,                                                           \
+    tuner->Run(dev_ctx,                                                       \
                key,                                                           \
                alpha,                                                         \
                beta,                                                          \
-               ctx,                                                           \
+               dev_ctx,                                                       \
                a,                                                             \
                b,                                                             \
                c,                                                             \
diff --git a/paddle/phi/kernels/sparse/gpu/sparse_attention_kernel.cu b/paddle/phi/kernels/sparse/gpu/sparse_attention_kernel.cu
@@ -206,7 +206,7 @@ input: sparse C in CSR format (num_rows,num_rows)
 output: sparse C after softmax operation
 */
 template <typename DeviceContext, typename T>
-void SparseSoftmaxForward(const phi::GPUContext& ctx,
+void SparseSoftmaxForward(const phi::GPUContext& dev_ctx,
                           const phi::DenseTensor* offset,
                           const phi::DenseTensor* columns,
                           phi::DenseTensor* input,
@@ -319,7 +319,7 @@ void SparseSoftmaxForward(const phi::GPUContext& ctx,
 }
 
 template <typename DeviceContext, typename T>
-void SparseSoftmaxBackward(const phi::GPUContext& ctx,
+void SparseSoftmaxBackward(const phi::GPUContext& dev_ctx,
                            const phi::DenseTensor* offset,
                            const phi::DenseTensor* columns,
                            phi::DenseTensor* dx,
@@ -450,7 +450,7 @@ input: dense A (num_rows,num_cols), dense B (num_rows,num_cols)
 output: sparse C in CSR format (num_rows,num_rows)
 */
 template <typename DeviceContext, typename T>
-void DotSdd(const phi::GPUContext& ctx,
+void DotSdd(const phi::GPUContext& dev_ctx,
             const phi::DenseTensor* a,
             const phi::DenseTensor* b,
             const phi::DenseTensor* c_offset,
@@ -519,9 +519,9 @@ void DotSdd(const phi::GPUContext& ctx,
                                          CUSPARSE_SDDMM_ALG_DEFAULT,
                                          &buffer_size);
   auto d_buffer_ptr = phi::memory_utils::Alloc(
-      ctx.GetPlace(),
+      dev_ctx.GetPlace(),
       buffer_size,
-      phi::Stream(reinterpret_cast<phi::StreamId>(ctx.stream())));
+      phi::Stream(reinterpret_cast<phi::StreamId>(dev_ctx.stream())));
   void* d_buffer = static_cast<void*>(d_buffer_ptr->ptr());
 
   phi::dynload::cusparseSDDMM(handle,
@@ -550,7 +550,7 @@ input: sparse A in CSR format (num_rows,num_rows), dense B (num_rows,num_cols)
 output: dense C (num_rows,num_cols)
 */
 template <typename DeviceContext, typename T>
-void DotDsd(const phi::GPUContext& ctx,
+void DotDsd(const phi::GPUContext& dev_ctx,
             const phi::DenseTensor* a_offset,
             const phi::DenseTensor* a_columns,
             const phi::DenseTensor* a_value,
@@ -621,9 +621,9 @@ void DotDsd(const phi::GPUContext& ctx,
                                         CUSPARSE_SPMM_ALG_DEFAULT,
                                         &buffer_size);
   auto d_buffer_ptr = phi::memory_utils::Alloc(
-      ctx.GetPlace(),
+      dev_ctx.GetPlace(),
       buffer_size,
-      phi::Stream(reinterpret_cast<phi::StreamId>(ctx.stream())));
+      phi::Stream(reinterpret_cast<phi::StreamId>(dev_ctx.stream())));
   void* d_buffer = static_cast<void*>(d_buffer_ptr->ptr());
 
   phi::dynload::cusparseSpMM(handle,
