Skip to content

P
PaddleDetection

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

通知 695
Star 11112
Fork 2696
P
PaddleDetection
提交 5dfce931 编写于 作者: G guoshengCS
浏览文件
操作

To make CUDA_LAUNCH_KERNEL_HELPER support large size. 

test=develop
上级 ba02ac46
显示空白变更内容
内联 并排
Showing with 6 addition and 4 deletion
paddle/fluid/platform/cuda_device_function.h
浏览文件 @ 5dfce931
...@@ -54,6 +54,8 @@ inline static int RoundToPowerOfTwo(int dim) { ...@@ -54,6 +54,8 @@ inline static int RoundToPowerOfTwo(int dim) {
} break } break
#define CUDA_LAUNCH_KERNEL_HELPER(...) \ #define CUDA_LAUNCH_KERNEL_HELPER(...) \
CUDA_LAUNCH_KERNEL_BASE(1024, ##__VA_ARGS__); \
CUDA_LAUNCH_KERNEL_BASE(512, ##__VA_ARGS__); \
CUDA_LAUNCH_KERNEL_BASE(256, ##__VA_ARGS__); \ CUDA_LAUNCH_KERNEL_BASE(256, ##__VA_ARGS__); \
CUDA_LAUNCH_KERNEL_BASE(128, ##__VA_ARGS__); \ CUDA_LAUNCH_KERNEL_BASE(128, ##__VA_ARGS__); \
CUDA_LAUNCH_KERNEL_BASE(64, ##__VA_ARGS__); \ CUDA_LAUNCH_KERNEL_BASE(64, ##__VA_ARGS__); \
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或