fix put along axis momory (#71863)

* fix put along axis momory

* fix bug

* update

* fix bug

* try fix bug

* fix compile bug

Author: phlrain

paddle/phi/kernels/stride/slice_grad_kernel.cc

@@ -16,6 +16,7 @@
 #include "paddle/common/flags.h"
 #include "paddle/phi/backends/all_context.h"
 #include "paddle/phi/core/kernel_registry.h"
+#include "paddle/phi/kernels/full_kernel.h"
 #include "paddle/phi/kernels/funcs/strided_utils.h"
 #include "paddle/phi/kernels/slice_kernel.h"
 
@@ -46,6 +47,16 @@ void SliceGradStridedKernel(const Context& dev_ctx,
                      }));
   DenseTensor tmp;
   tmp.set_meta(out_grad.meta());
+  if (out_grad.numel() == 0) {
+    // set zero to input_grad
+
+    PD_VISIT_ALL_TYPES(input.dtype(), "SliceGradStridedKernel", ([&] {
+                         phi::StridedTensorFill<data_t>(
+                             *input_grad, 0, input_grad);
+                       }));
+
+    return;
+  }
   SliceStridedKernel<Context>(dev_ctx,
                               *input_grad,
                               axes,