fix gather compile error

PaddlePaddle · Sep 7, 2021 · c7f8dce · c7f8dce · paddle-bot-old · Sep 8, 2021
1 parent fbaa127
commit c7f8dce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/fluid/operators/gather.cu.h b/paddle/fluid/operators/gather.cu.h
@@ -42,7 +42,7 @@ __global__ void GatherCUDAKernel(const T* params, const IndexT* indices,
 }
 
 template <typename T, typename IndexT = int>
-__global__ void GatherNdCUDAKernel(const T* input, const int* input_dims,
+__global__ void GatherNdCUDAKernel(const T* input, const int64_t* input_dims,
                                    const IndexT* indices, T* output,
                                    size_t remain_size, size_t slice_size,
                                    size_t end_size) {