NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-04-20 14:58:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add cupti path for ubuntu packages.

Browse Source 
Fixes #59
This commit is contained in:
Allison Vacanti
2021-12-20 14:16:53 -05:00
parent ff1ad78cfa
commit 61d094abf1
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
cmake/NVBenchCUPTI.cmake
View File

@@ -4,7 +4,13 @@ if (NOT TARGET CUDA::toolkit)
find_package(CUDAToolkit REQUIRED)
endif()
set(nvbench_cupti_root "${CUDAToolkit_LIBRARY_ROOT}/extras/CUPTI")
if (EXISTS "${CUDAToolkit_LIBRARY_ROOT}/extras/CUPTI/lib64")
# NVIDIA installer layout:
set(nvbench_cupti_root "${CUDAToolkit_LIBRARY_ROOT}/extras/CUPTI")
else()
# Ubuntu package layout:
set(nvbench_cupti_root "${CUDAToolkit_LIBRARY_ROOT}")
endif()
# The CUPTI targets in FindCUDAToolkit are broken:
# - The dll locations are not specified