Add NVML support for persistence mode, locking clocks.

Locking clocks is currently only implemented for Volta+ devices.

Example usage:

my_bench -d [0,1,3] --persistence-mode 1 --lock-gpu-clocks base

See the cli_help.md docs for more info.
This commit is contained in:
Allison Vacanti
2021-10-21 01:01:16 -04:00
parent d0c90ff920
commit b948e79cab
18 changed files with 656 additions and 19 deletions

View File

@@ -57,3 +57,10 @@ rapids_find_package(CUDAToolkit REQUIRED
# Append CTK targets to this as we add optional deps (NMVL, CUPTI, ...)
set(ctk_libraries CUDA::toolkit)
################################################################################
# CUDAToolkit -> NVML
if (NVBench_ENABLE_NVML)
include("${CMAKE_CURRENT_LIST_DIR}/NVBenchNVML.cmake")
list(APPEND ctk_libraries nvbench::nvml)
endif()