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) ...@@ -86,7 +86,8 @@ if(USE_CUDA)
-DCUDA_HAS_FP16=1 \ -DCUDA_HAS_FP16=1 \
-D__CUDA_NO_HALF_OPERATORS__ \ -D__CUDA_NO_HALF_OPERATORS__ \
-D__CUDA_NO_HALF_CONVERSIONS__ \ -D__CUDA_NO_HALF_CONVERSIONS__ \
-D__CUDA_NO_HALF2_OPERATORS__" -D__CUDA_NO_HALF2_OPERATORS__ \
--gpu-max-threads-per-block=1024"
) )
endif() 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