Merge branch 'cnet' of https://github.com/ceccocats/tkDNN into cnet

2020-02-11 10:32:52 +01:00
parent 73548b179e 2b06836a7f
commit e01d7c6b13
3 changed files with 180 additions and 60 deletions
@@ -17,6 +17,7 @@ endif()
 find_package(CUDA 9.0 REQUIRED)
 SET(CUDA_SEPARABLE_COMPILATION ON)
 #set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -arch=sm_30 --compiler-options '-fPIC'")
+#set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS} --maxrregcount=32)

 find_package(CUDNN REQUIRED)