Skip to content

Commit fa71fee

Browse files
authored
[PHI] Fix full_like kernel for big tensor (#72831)
1 parent 571146e commit fa71fee

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/phi/kernels/gpu/full_kernel.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ void FullLikeKernel(const Context& dev_ctx,
6969
// This function has no input, so the inputs.size() == 0. Use kUnary, but the
7070
// data will not be loaded in the kernel because the number of parameters in
7171
// the operator is 0
72-
int numel = out->numel();
72+
int64_t numel = out->numel();
7373

7474
if (!std::is_same<T, phi::dtype::complex<float>>::value &&
7575
!std::is_same<T, phi::dtype::complex<double>>::value) {

0 commit comments

Comments
 (0)