replace 25 with phi::DDim::kMaxRank+1

Author: LittleHeroZZZX

paddle/phi/kernels/cpu/index_elementwise_get_grad_kernel.cc

@@ -27,8 +27,8 @@ void IndexEleGetGradAccKernel(
     const char* in_ptr,
     char* out_ptr,
     const std::array<char*, DDim::kMaxRank> index_ptrs,
-    const std::array<int64_t, 25> sizes,
-    const std::array<int64_t, 25> strides,
+    const std::array<int64_t, phi::DDim::kMaxRank + 1> sizes,
+    const std::array<int64_t, phi::DDim::kMaxRank + 1> strides,
     int num_indices,
     offset_calc_t offset_calc) {
   for (int64_t idx = 0; idx < N; idx++) {
@@ -60,8 +60,8 @@ void CPUIndexElementwiseGetGrad(const phi::CPUContext& ctx,
                                 DenseTensor* output) {
   int64_t numel = 0;
   auto num_indices = index_dims.size();
-  auto sizes = std::array<int64_t, 25>{};
-  auto strides = std::array<int64_t, 25>{};
+  auto sizes = std::array<int64_t, phi::DDim::kMaxRank + 1>{};
+  auto strides = std::array<int64_t, phi::DDim::kMaxRank + 1>{};
   for (unsigned i = 0; i < num_indices; i++) {
     sizes[i] = index_dims[i];
     strides[i] = index_strides[i];

paddle/phi/kernels/cpu/index_elementwise_put_grad_kernel.cc

@@ -40,8 +40,8 @@ void CPUIndexElementwisePutGradKernel(
     DenseTensor* value_grad) {
   int64_t numel = 0;
   auto num_indices = index_dims.size();
-  auto sizes = std::array<int64_t, 25>{};
-  auto strides = std::array<int64_t, 25>{};
+  auto sizes = std::array<int64_t, phi::DDim::kMaxRank + 1>{};
+  auto strides = std::array<int64_t, phi::DDim::kMaxRank + 1>{};
   for (unsigned i = 0; i < num_indices; i++) {
     sizes[i] = index_dims[i];
     strides[i] = index_strides[i];

paddle/phi/kernels/cpu/index_elementwise_put_kernel.cc

@@ -44,8 +44,8 @@ void CPUIndexElementwisePutKernel(const phi::CPUContext& dev_ctx,
     phi::Copy(dev_ctx, input, dev_ctx.GetPlace(), false, output);
   }
   auto num_indices = index_dims.size();
-  auto sizes = std::array<int64_t, 25>{};
-  auto strides = std::array<int64_t, 25>{};
+  auto sizes = std::array<int64_t, phi::DDim::kMaxRank + 1>{};
+  auto strides = std::array<int64_t, phi::DDim::kMaxRank + 1>{};
   for (unsigned i = 0; i < num_indices; i++) {
     sizes[i] = index_dims[i];
     strides[i] = index_strides[i];

paddle/phi/kernels/gpu/index_elementwise_get_grad_kernel.cu

@@ -29,8 +29,8 @@ __global__ void IndexEleGetGradAccKernel(
     const char* in_ptr,
     char* out_ptr,
     const std::array<char*, DDim::kMaxRank> index_ptrs,
-    const std::array<int64_t, 25> sizes,
-    const std::array<int64_t, 25> strides,
+    const std::array<int64_t, phi::DDim::kMaxRank + 1> sizes,
+    const std::array<int64_t, phi::DDim::kMaxRank + 1> strides,
     int num_indices,
     offset_calc_t offset_calc) {
   const int tid = threadIdx.x;
@@ -74,8 +74,8 @@ void GPUIndexElementwiseGetGrad(const phi::GPUContext& ctx,
 
   auto num_indices = index_dims.size();
 
-  auto sizes = std::array<int64_t, 25>{};
-  auto strides = std::array<int64_t, 25>{};
+  auto sizes = std::array<int64_t, phi::DDim::kMaxRank>{};
+  auto strides = std::array<int64_t, phi::DDim::kMaxRank>{};
   for (unsigned i = 0; i < num_indices; i++) {
     sizes[i] = index_dims[i];
     strides[i] = index_strides[i];

paddle/phi/kernels/gpu/index_elementwise_put_grad_kernel.cu

@@ -43,8 +43,8 @@ void GPUIndexElementwisePutGradKernel(
 
   auto num_indices = index_dims.size();
 
-  auto sizes = std::array<int64_t, 25>{};
-  auto strides = std::array<int64_t, 25>{};
+  auto sizes = std::array<int64_t, phi::DDim::kMaxRank + 1>{};
+  auto strides = std::array<int64_t, phi::DDim::kMaxRank + 1>{};
   for (unsigned i = 0; i < num_indices; i++) {
     sizes[i] = index_dims[i];
     strides[i] = index_strides[i];

paddle/phi/kernels/gpu/index_elementwise_put_kernel.cu

@@ -37,8 +37,8 @@ void GPUIndexElementwisePutKernel(const phi::GPUContext& dev_ctx,
 
   auto num_indices = index_dims.size();
 
-  auto sizes = std::array<int64_t, 25>{};
-  auto strides = std::array<int64_t, 25>{};
+  auto sizes = std::array<int64_t, phi::DDim::kMaxRank + 1>{};
+  auto strides = std::array<int64_t, phi::DDim::kMaxRank + 1>{};
   for (unsigned i = 0; i < num_indices; i++) {
     sizes[i] = index_dims[i];
     strides[i] = index_strides[i];