diff options
| author | Maxime Schmitt <maxime.schmitt@etu.unistra.fr> | 2018-04-07 23:49:15 +0200 |
|---|---|---|
| committer | Maxime Schmitt <maxime.schmitt@etu.unistra.fr> | 2018-04-07 23:49:15 +0200 |
| commit | cea46dea9578fc0a00b9e89edaf718ef0754b37c (patch) | |
| tree | 492cf8506597fccde6ef0d58360eea66a17102a2 /cmake | |
| parent | 455a70b38ec438cb9d40d1c808b011daeed1be26 (diff) | |
Architecture aware regex search for nvml library
Diffstat (limited to 'cmake')
| -rw-r--r-- | cmake/modules/FindNVML.cmake | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/cmake/modules/FindNVML.cmake b/cmake/modules/FindNVML.cmake index c1763f7..a546355 100644 --- a/cmake/modules/FindNVML.cmake +++ b/cmake/modules/FindNVML.cmake @@ -24,7 +24,11 @@ find_path(NVML_INCLUDE_DIRS NAMES nvml.h PATHS ${nvml_header_path_hint} ${PROJECT_BINARY_DIR}/include) # library -file(GLOB nvml_lib_path_hint /usr/lib*/nvidia*/) +if("${CMAKE_SIZEOF_VOID_P}" EQUAL "8") # 64bit + file(GLOB nvml_lib_path_hint /usr/lib64/nvidia*/ /usr/lib/nvidia*/) +else() # assume 32bit + file(GLOB nvml_lib_path_hint /usr/lib32/nvidia*/ /usr/lib/nvidia*/) +endif() find_library(NVML_LIBRARIES NAMES nvidia-ml libnvidia-ml.so.1 PATHS ${nvml_lib_path_hint}) |