[fluid_ops] Fix typos in c_softmax_with_cross_entropy_grad_kernel.cu (#70344)

co63oc · web-flow · commit 9384f996846c · 2024-12-31T16:53:47.000+08:00
diff --git a/paddle/phi/kernels/gpu/c_softmax_with_cross_entropy_grad_kernel.cu b/paddle/phi/kernels/gpu/c_softmax_with_cross_entropy_grad_kernel.cu
@@ -121,10 +121,10 @@ void CSoftmaxWithCrossEntropyGradKernel(const Context& dev_ctx,
   if (logit_grad != softmax) {
     phi::Copy(dev_ctx, *softmax, dev_ctx.GetPlace(), false, logit_grad);
   }
-  const auto sofrmax_dims = softmax->dims();
-  const int axis = sofrmax_dims.size() - 1;
-  const int64_t N = phi::funcs::SizeToAxis<int64_t>(axis, sofrmax_dims);
-  const int64_t D = phi::funcs::SizeFromAxis<int64_t>(axis, sofrmax_dims);
+  const auto softmax_dims = softmax->dims();
+  const int axis = softmax_dims.size() - 1;
+  const int64_t N = phi::funcs::SizeToAxis<int64_t>(axis, softmax_dims);
+  const int64_t D = phi::funcs::SizeFromAxis<int64_t>(axis, softmax_dims);
 
   const auto label_dims = labels->dims();
   const int64_t C = label_dims[axis];
diff --git a/paddle/phi/kernels/gpu/c_softmax_with_multi_label_cross_entropy_grad_kernel.cu b/paddle/phi/kernels/gpu/c_softmax_with_multi_label_cross_entropy_grad_kernel.cu
@@ -92,10 +92,10 @@ void CSoftmaxWithMultiLabelCrossEntropyGradKernel(
   if (logit_grad != softmax) {
     phi::Copy(dev_ctx, *softmax, dev_ctx.GetPlace(), false, logit_grad);
   }
-  const auto sofrmax_dims = softmax->dims();
-  const int axis = sofrmax_dims.size() - 1;
-  const int64_t N = phi::funcs::SizeToAxis<int64_t>(axis, sofrmax_dims);
-  const int64_t D = phi::funcs::SizeFromAxis<int64_t>(axis, sofrmax_dims);
+  const auto softmax_dims = softmax->dims();
+  const int axis = softmax_dims.size() - 1;
+  const int64_t N = phi::funcs::SizeToAxis<int64_t>(axis, softmax_dims);
+  const int64_t D = phi::funcs::SizeFromAxis<int64_t>(axis, softmax_dims);
 
   const auto label_dims = labels->dims();
   const int64_t C = label_dims[axis];
