Commit 8f2f74c3 authored by zhanggzh's avatar zhanggzh
Browse files

add --gpu-max-threads-per-block

parent d79540ef
......@@ -86,7 +86,8 @@ if(USE_CUDA)
-DCUDA_HAS_FP16=1 \
-D__CUDA_NO_HALF_OPERATORS__ \
-D__CUDA_NO_HALF_CONVERSIONS__ \
-D__CUDA_NO_HALF2_OPERATORS__"
-D__CUDA_NO_HALF2_OPERATORS__ \
--gpu-max-threads-per-block=1024"
)
endif()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment