Skip to content

P
Paddle

PaddlePaddle / Paddle
大约 1 年 前同步成功

通知 2299
Star 20931
Fork 5422
P
Paddle
未验证 提交 8b58862a 编写于 作者: W Wilber 提交者： GitHub
浏览文件
操作

fix compile error on jetson (#39441)

上级 52d6b306
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
paddle/pten/backends/gpu/gpu_launch_config.h
浏览文件 @ 8b58862a
...@@ -64,7 +64,7 @@ static inline int RoundToPowerOfTwo(int n) { ...@@ -64,7 +64,7 @@ static inline int RoundToPowerOfTwo(int n) {
#ifdef WITH_NV_JETSON #ifdef WITH_NV_JETSON
// The number of threads cannot be assigned 1024 in some cases when the device // The number of threads cannot be assigned 1024 in some cases when the device
// is nano or tx2 . // is nano or tx2 .
inline void ChangeThreadNum(const pten::CUDAContext& context, inline void ChangeThreadNum(const pten::GPUContext& context,
int* num_thread, int* num_thread,
int alternative_num_thread = 512) { int alternative_num_thread = 512) {
if (context.GetComputeCapability() == 53 || if (context.GetComputeCapability() == 53 ||
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或