From cfaee812187ade614dc7797856b2e3154e76af3c Mon Sep 17 00:00:00 2001 From: chalsliu <45041955+chalsliu@users.noreply.github.com> Date: Mon, 5 Sep 2022 11:17:58 +0800 Subject: [PATCH] Fix jetson compile error (#45692) --- paddle/fluid/platform/device/gpu/gpu_launch_config.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/platform/device/gpu/gpu_launch_config.h b/paddle/fluid/platform/device/gpu/gpu_launch_config.h index ca861d543f1..d253a92c986 100644 --- a/paddle/fluid/platform/device/gpu/gpu_launch_config.h +++ b/paddle/fluid/platform/device/gpu/gpu_launch_config.h @@ -68,8 +68,8 @@ static inline int RoundToPowerOfTwo(int n) { #ifdef WITH_NV_JETSON // The number of threads cannot be assigned 1024 in some cases when the device // is nano or tx2 . -template -inline void ChangeThreadNum(const phi::GPUContext& context, +template +inline void ChangeThreadNum(const GPUContext& context, int* num_thread, int alternative_num_thread = 512) { if (context.GetComputeCapability() == 53 || -- GitLab