wehub-resource-sync
56 lines
1.3 KiB
CMake
56 lines
1.3 KiB
CMake
if((NOT WITH_GPU)
|
|
OR WIN32
|
|
OR APPLE)
|
|
set(NVTX_FOUND OFF)
|
|
return()
|
|
endif()
|
|
|
|
include(${PROJECT_SOURCE_DIR}/cmake/architecture.cmake)
|
|
|
|
set(NVTX_ROOT
|
|
"/usr"
|
|
CACHE PATH "NVTX ROOT")
|
|
find_path(
|
|
NVTX_INCLUDE_DIR nvToolsExt.h
|
|
PATHS ${NVTX_ROOT} ${NVTX_ROOT}/include $ENV{NVTX_ROOT}
|
|
$ENV{NVTX_ROOT}/include ${CUDA_TOOLKIT_INCLUDE}
|
|
NO_DEFAULT_PATH)
|
|
|
|
get_filename_component(__libpath_hint ${CUDA_CUDART_LIBRARY} PATH)
|
|
|
|
paddle_normalize_target_arch(TARGET_ARCH)
|
|
paddle_detect_cuda_target_dir(CUDA_TARGET_DIR)
|
|
|
|
list(
|
|
APPEND
|
|
NVTX_CHECK_LIBRARY_DIRS
|
|
${NVTX_ROOT}
|
|
${NVTX_ROOT}/lib64
|
|
${NVTX_ROOT}/lib
|
|
${NVTX_ROOT}/lib/${TARGET_ARCH}-linux-gnu
|
|
$ENV{NVTX_ROOT}
|
|
$ENV{NVTX_ROOT}/lib64
|
|
$ENV{NVTX_ROOT}/lib
|
|
${CUDA_TOOLKIT_ROOT_DIR}
|
|
${CUDA_TOOLKIT_ROOT_DIR}/targets/${TARGET_ARCH}-linux/lib
|
|
${CUDA_TOOLKIT_ROOT_DIR}/targets/${CUDA_TARGET_DIR}/lib64
|
|
${CUDA_TOOLKIT_ROOT_DIR}/targets/${CUDA_TARGET_DIR}/lib)
|
|
|
|
find_library(
|
|
CUDA_NVTX_LIB
|
|
NAMES libnvToolsExt.so
|
|
PATHS ${NVTX_CHECK_LIBRARY_DIRS} ${NVTX_INCLUDE_DIR} ${__libpath_hint}
|
|
NO_DEFAULT_PATH
|
|
DOC "Path to the NVTX library.")
|
|
|
|
if(NVTX_INCLUDE_DIR AND CUDA_NVTX_LIB)
|
|
set(NVTX_FOUND ON)
|
|
else()
|
|
set(NVTX_FOUND OFF)
|
|
endif()
|
|
|
|
if(NVTX_FOUND)
|
|
include_directories(${NVTX_INCLUDE_DIR})
|
|
add_definitions(-DCINN_WITH_NVTX)
|
|
endif()
|