Fixing issue #1395: Undefined symbol: cudaGetDevice (#1396)

8a90f877 · Sammy · Tianqi Chen · 4dd41ddb · 8a90f877
Commit 8a90f877 authored Jul 06, 2018 by Sammy Committed by Tianqi Chen Jul 06, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

cmake/util/FindCUDA.cmake
+2 -2

No files found.
--- a/cmake/util/FindCUDA.cmake
+++ b/cmake/util/FindCUDA.cmake
@@ -28,11 +28,11 @@ macro(find_cuda use_cuda)
    set(CUDA_INCLUDE_DIRS ${CUDA_TOOLKIT_ROOT_DIR}/include)
    set(CUDA_FOUND TRUE)
    if(MSVC)
-      find_library(CUDA_CUDAT_LIBRARY cudart
+      find_library(CUDA_CUDART_LIBRARY cudart
        ${CUDA_TOOLKIT_ROOT_DIR}/lib64
        ${CUDA_TOOLKIT_ROOT_DIR}/lib)
    else(MSVC)
-      find_library(CUDA_CUDAT_LIBRARY cudart
+      find_library(CUDA_CUDART_LIBRARY cudart
        ${CUDA_TOOLKIT_ROOT_DIR}/lib64
        ${CUDA_TOOLKIT_ROOT_DIR}/lib)
    endif(MSVC)