diff --git a/NVML/cmake/FindNVML.cmake b/NVML/cmake/FindNVML.cmake index 3cf7b23..f2c5fb5 100644 --- a/NVML/cmake/FindNVML.cmake +++ b/NVML/cmake/FindNVML.cmake @@ -45,17 +45,16 @@ # ############################################################################### -#TODO: Should GPU_DEPLOYMENT_KIT_ROOT_DIR override the default locations? if( CMAKE_SYSTEM_NAME STREQUAL "Windows" ) - set( NVML_LIB_PATHS "C:\Program Files\NVIDIA Corporation\GDK" ) + set( NVML_LIB_PATHS "C:/Program Files/NVIDIA Corporation/GDK/nvml/lib" ) if(GPU_DEPLOYMENT_KIT_ROOT_DIR) - list(APPEND NVML_LIB_PATHS "${GPU_DEPLOYMENT_KIT_ROOT_DIR}\nvml\lib") + list(APPEND NVML_LIB_PATHS "${GPU_DEPLOYMENT_KIT_ROOT_DIR}/nvml/lib") endif() set(NVML_NAMES nvml) - set( NVML_INC_PATHS "C:\Program Files\NVIDIA Corporation\GDK\nvml\include" ) + set( NVML_INC_PATHS "C:/Program Files/NVIDIA Corporation/GDK/nvml/include" ) if(GPU_DEPLOYMENT_KIT_ROOT_DIR) - list(APPEND NVML_INC_PATHS "${GPU_DEPLOYMENT_KIT_ROOT_DIR}\nvml\include") + list(APPEND NVML_INC_PATHS "${GPU_DEPLOYMENT_KIT_ROOT_DIR}/nvml/include") endif() else() set( NVML_LIB_PATHS /usr/lib64 ) @@ -76,7 +75,7 @@ find_path(NVML_INCLUDE_DIR nvml.h PATHS ${NVML_INC_PATHS}) # handle the QUIETLY and REQUIRED arguments and set NVML_FOUND to TRUE if # all listed variables are TRUE -include(${CMAKE_CURRENT_LIST_DIR}/FindPackageHandleStandardArgs.cmake) +include(FindPackageHandleStandardArgs) FIND_PACKAGE_HANDLE_STANDARD_ARGS(NVML DEFAULT_MSG NVML_LIBRARY NVML_INCLUDE_DIR) mark_as_advanced(NVML_LIBRARY NVML_INCLUDE_DIR)