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 an nvbench-ctl executable.

Browse Source 
This will provide functionality such as clock locking (--lgm),
persistance mode (--pm), device querying (--list), version checking
(--version), and documentation (--help).

This is possible already with any nvbench executable, but having
one with a reliable name will be helpful for scripting and writing
documentation.
This commit is contained in:
Allison Vacanti
2021-12-20 12:04:31 -05:00
parent a8422197a9
commit 20522c807d
9 changed files with 160 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
testing/cmake/CMakeLists.txt
View File

@@ -47,6 +47,7 @@ nvbench_add_compile_test(test_name
test_export
build_tree
-D "NVBench_DIR=${NVBench_BINARY_DIR}"
-D TEST_TYPE=BUILD_TREE
)
################################################################################
@@ -57,6 +58,7 @@ nvbench_add_compile_test(test_name
install_tree
# "rapids_export() always installs to lib" per rapids_export docs
-D "NVBench_DIR=${tmp_install_prefix}/lib/cmake/nvbench/"
-D TEST_TYPE=INSTALL_TREE
)
set_tests_properties(${test_name} PROPERTIES FIXTURES_REQUIRED install_tree)

48
testing/cmake/test_export/CMakeLists.txt
View File

@@ -8,11 +8,47 @@ add_executable(test_bench test_bench.cu)
target_link_libraries(test_bench PRIVATE nvbench::main)
enable_testing()
add_test(NAME test_bench COMMAND "$<TARGET_FILE:test_bench>" --timeout 1)
add_test(NAME nvbench_ctl COMMAND "$<TARGET_FILE:nvbench::ctl>")
# Setup LD_LIBRARY_PATH for testing
if (UNIX)
set(ctl_lib_path "")
set(cupti_lib_path "")
# Need to find installed libnvbench.so for installed nvbench-ctl.
# Not needed for build_tree test because of RUNPATH.
if (TEST_TYPE STREQUAL "INSTALL_TREE")
get_property(nvbench_config TARGET nvbench::nvbench
PROPERTY IMPORTED_CONFIGURATIONS
)
list(LENGTH nvbench_config num_configs)
if (num_configs GREATER 1)
message(WARNING
"Multiple IMPORTED_CONFIGURATIONS for nvbench::nvbench. "
"Picking the first one. This may cause issues."
)
list(GET nvbench_config 0 nvbench_config)
endif()
get_property(ctl_lib_path TARGET nvbench::nvbench
PROPERTY IMPORTED_LOCATION_${nvbench_config}
)
cmake_path(GET ctl_lib_path PARENT_PATH ctl_lib_path)
endif()
# Need to add the CUPTI path to LD_LIBRARY_PATH to make sure CUPTI libraries
# are found at runtime:
if (TARGET nvbench::cupti)
get_property(cupti_lib_path TARGET nvbench::cupti PROPERTY IMPORTED_LOCATION)
cmake_path(GET cupti_lib_path PARENT_PATH cupti_lib_path)
endif()
set_property(TEST test_bench PROPERTY
ENVIRONMENT "LD_LIBRARY_PATH=${cupti_lib_path}"
)
set_property(TEST nvbench_ctl PROPERTY
ENVIRONMENT "LD_LIBRARY_PATH=${ctl_lib_path}:${cupti_lib_path}"
)
# Need to add the CUPTI path to LD_LIBRARY_PATH to make sure CUPTI libraries
# are found at runtime:
if (UNIX AND TARGET nvbench::cupti)
get_property(cupti_lib_path TARGET nvbench::cupti PROPERTY IMPORTED_LOCATION)
cmake_path(GET cupti_lib_path PARENT_PATH cupti_lib_path)
set_property(TEST test_bench PROPERTY ENVIRONMENT "LD_LIBRARY_PATH=${cupti_lib_path}")
endif()