To make CUDA_LAUNCH_KERNEL_HELPER support large size.

test=develop
6 years ago · 5dfce93101
parent ba02ac4692
commit 5dfce93101
1 changed files with 6 additions and 4 deletions
--- a/paddle/fluid/platform/cuda_device_function.h
+++ b/paddle/fluid/platform/cuda_device_function.h
@ -54,6 +54,8 @@ inline static int RoundToPowerOfTwo(int dim) {
  } break

 #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(128, ##__VA_ARGS__);  \
  CUDA_LAUNCH_KERNEL_BASE(64, ##__VA_ARGS__);   \