From b51bf77195d2af7b5321d6f546dc3872fb39c3fc Mon Sep 17 00:00:00 2001 From: Liang Shuhao Date: Wed, 21 May 2025 03:01:06 +0000 Subject: [PATCH] [PHI] Fix full_like kernel for big tensor --- paddle/phi/kernels/gpu/full_kernel.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/phi/kernels/gpu/full_kernel.cu b/paddle/phi/kernels/gpu/full_kernel.cu index fccbaa6b9e2917..cd5b6a2fb4847b 100644 --- a/paddle/phi/kernels/gpu/full_kernel.cu +++ b/paddle/phi/kernels/gpu/full_kernel.cu @@ -69,7 +69,7 @@ void FullLikeKernel(const Context& dev_ctx, // This function has no input, so the inputs.size() == 0. Use kUnary, but the // data will not be loaded in the kernel because the number of parameters in // the operator is 0 - int numel = out->numel(); + int64_t numel = out->numel(); if (!std::is_same>::value && !std::is_same>::value) {