diff --git a/paddle/phi/kernels/gpu/contiguous_kernel.cu b/paddle/phi/kernels/gpu/contiguous_kernel.cu index d259fb476f03f9..63a6c0cc23fcc7 100644 --- a/paddle/phi/kernels/gpu/contiguous_kernel.cu +++ b/paddle/phi/kernels/gpu/contiguous_kernel.cu @@ -287,7 +287,7 @@ bool LaunchContiguousCazeOneKernel( const phi::Array& input_stride, const phi::Array& input_dims, int rank, - int numel, + int64_t numel, T* output_data) { dim3 grid(1, 1, 1), block(1, 1, 1); phi::Array cur_input_dims; @@ -427,7 +427,7 @@ void LaunchContiguousDefaultKernel( const phi::Array& input_stride, const phi::Array& input_dims, int rank, - int numel, + int64_t numel, T* output_data) { int64_t block = 512; int64_t grid = (numel + block - 1) / block;