diff --git a/src/dark_cuda.c b/src/dark_cuda.c index 8fdaee3f..d155b408 100644 --- a/src/dark_cuda.c +++ b/src/dark_cuda.c @@ -18,14 +18,13 @@ int gpu_index = 0; #include #include +#ifndef USE_CMAKE_LIBS #pragma comment(lib, "cuda.lib") - #ifdef CUDNN -#ifndef USE_CMAKE_LIBS #pragma comment(lib, "cudnn.lib") -#endif // USE_CMAKE_LIBS #endif // CUDNN +#endif // USE_CMAKE_LIBS #if defined(CUDNN_HALF) && !defined(CUDNN) #error "If you set CUDNN_HALF=1 then you must set CUDNN=1"