fix compile error on jetson (#39441)

PaddlePaddle · Feb 10, 2022 · 8b58862 · 8b58862
1 parent 52d6b30
commit 8b58862
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/pten/backends/gpu/gpu_launch_config.h b/paddle/pten/backends/gpu/gpu_launch_config.h
@@ -64,7 +64,7 @@ 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 .
-inline void ChangeThreadNum(const pten::CUDAContext& context,
+inline void ChangeThreadNum(const pten::GPUContext& context,
                             int* num_thread,
                             int alternative_num_thread = 512) {
   if (context.GetComputeCapability() == 53 ||