27 lines
581 B
CMake
27 lines
581 B
CMake
if (NVTX_LIBRARY)
|
|
unset(NVTX_LIBRARY CACHE)
|
|
endif (NVTX_LIBRARY)
|
|
|
|
set(NVTX_LIB_NAME nvToolsExt)
|
|
|
|
|
|
find_path(NVTX_INCLUDE_DIR
|
|
NAMES nvToolsExt.h
|
|
PATHS ${CUDA_HOME}/include ${CUDA_INCLUDE} /usr/local/cuda/include)
|
|
|
|
|
|
find_library(NVTX_LIBRARY
|
|
NAMES nvToolsExt
|
|
PATHS ${CUDA_HOME}/lib64 /usr/local/cuda/lib64)
|
|
|
|
message(STATUS "Using nvtx library: ${NVTX_LIBRARY}")
|
|
|
|
include(FindPackageHandleStandardArgs)
|
|
find_package_handle_standard_args(NVTX DEFAULT_MSG
|
|
NVTX_INCLUDE_DIR NVTX_LIBRARY)
|
|
|
|
mark_as_advanced(
|
|
NVTX_INCLUDE_DIR
|
|
NVTX_LIBRARY
|
|
)
|