cmake cudnn fix

2020-06-14 12:57:29 +02:00
parent ab6d2d1766
commit 567dc0f75d
2 changed files with 64 additions and 29 deletions
@@ -20,6 +20,8 @@ SET(CUDA_SEPARABLE_COMPILATION ON)
 set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS} --maxrregcount=32)

 find_package(CUDNN REQUIRED)
+include_directories(${CUDNN_INCLUDE_DIR})
+

 # compile
 file(GLOB tkdnn_CUSRC "src/kernels/*.cu" "src/sorting.cu")