Fix

co63oc · co63oc · commit 9366442b8b62 · 2025-05-29T08:44:50.000+08:00
diff --git a/paddle/phi/kernels/cpu/kthvalue_kernel.cc b/paddle/phi/kernels/cpu/kthvalue_kernel.cc
@@ -85,6 +85,7 @@ void KthvalueKernel(const Context& dev_ctx,
         dev_ctx, phi::IntArray(common::vectorize(output->dims())), NAN, output);
     phi::Full<int64_t, Context>(
         dev_ctx, phi::IntArray(common::vectorize(indices->dims())), 0, indices);
+    return;
   }
   const auto& in_dims = x.dims();
   if (axis < 0) axis += in_dims.size();
diff --git a/paddle/phi/kernels/gpu/kthvalue_kernel.cu b/paddle/phi/kernels/gpu/kthvalue_kernel.cu
@@ -166,6 +166,7 @@ void KthvalueKernel(const Context& dev_ctx,
         dev_ctx, phi::IntArray(common::vectorize(output->dims())), NAN, output);
     phi::Full<int64_t, Context>(
         dev_ctx, phi::IntArray(common::vectorize(indices->dims())), 0, indices);
+    return;
   }
 
   const auto& in_dims = x.dims();

Original file line number	Diff line number	Diff line change
`@@ -85,6 +85,7 @@ void KthvalueKernel(const Context& dev_ctx,`
`85`	`85`	`dev_ctx, phi::IntArray(common::vectorize(output->dims())), NAN, output);`
`86`	`86`	`phi::Full<int64_t, Context>(`
`87`	`87`	`dev_ctx, phi::IntArray(common::vectorize(indices->dims())), 0, indices);`
	`88`	`+ return;`
`88`	`89`	`}`
`89`	`90`	`const auto& in_dims = x.dims();`
`90`	`91`	`if (axis < 0) axis += in_dims.size();`
Original file line number	Diff line number	Diff line change
`@@ -166,6 +166,7 @@ void KthvalueKernel(const Context& dev_ctx,`
`166`	`166`	`dev_ctx, phi::IntArray(common::vectorize(output->dims())), NAN, output);`
`167`	`167`	`phi::Full<int64_t, Context>(`
`168`	`168`	`dev_ctx, phi::IntArray(common::vectorize(indices->dims())), 0, indices);`
	`169`	`+ return;`
`169`	`170`	`}`
`170`	`171`
`171`	`172`	`const auto& in_dims = x.dims();`