diff --git a/paddle/fluid/platform/device/gpu/gpu_launch_config.h b/paddle/fluid/platform/device/gpu/gpu_launch_config.h
index ca861d543f1ef16d0e83ef26bad068cea2770fde..d253a92c986cee8f405ce1942667d3ac20d8086d 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 <typename phi::GPUContext>
-inline void ChangeThreadNum(const phi::GPUContext& context,
+template <typename GPUContext>
+inline void ChangeThreadNum(const GPUContext& context,
                             int* num_thread,
                             int alternative_num_thread = 512) {
   if (context.GetComputeCapability() == 53 ||