diff --git a/paddle/phi/kernels/stride/strided_slice_grad_kernel.cc b/paddle/phi/kernels/stride/strided_slice_grad_kernel.cc index 0b4738509037f7..beb2ad5e294c23 100644 --- a/paddle/phi/kernels/stride/strided_slice_grad_kernel.cc +++ b/paddle/phi/kernels/stride/strided_slice_grad_kernel.cc @@ -44,7 +44,7 @@ void StridedSliceRawGradStridedKernel(const Context& dev_ctx, PD_VISIT_ALL_TYPES(x_grad->dtype(), "StridedSliceRawGradStridedKernel", ([&] { phi::StridedTensorFill(*x_grad, 0, x_grad); })); - if (x_grad->numel() == 0) return; + if (out_grad.numel() == 0) return; DenseTensor tmp; tmp.set_layout(out_grad.layout()); tmp.set_lod(out_grad.lod());