if(NOT WITH_GPU AND NOT WITH_ROCM)

return()

endif()

if(WITH_ROCM)

set(CUPTI_ROOT

"${ROCM_PATH}/cuda/extras/CUPTI"

CACHE PATH "CUPTI ROOT")

else()

set(CUPTI_ROOT

"/usr"

CACHE PATH "CUPTI ROOT")

endif()

find_path(

CUPTI_INCLUDE_DIR cupti.h

PATHS ${CUPTI_ROOT}

${CUPTI_ROOT}/include

$ENV{CUPTI_ROOT}

$ENV{CUPTI_ROOT}/include

${CUDA_TOOLKIT_ROOT_DIR}/extras/CUPTI/include

${CUDA_TOOLKIT_ROOT_DIR}/targets/x86_64-linux/include

${CUDA_TOOLKIT_ROOT_DIR}/targets/aarch64-linux/include

NO_DEFAULT_PATH)

get_filename_component(__libpath_hist ${CUDA_CUDART_LIBRARY} PATH)

set(TARGET_ARCH "x86_64")

if(NOT ${CMAKE_SYSTEM_PROCESSOR})

set(TARGET_ARCH ${CMAKE_SYSTEM_PROCESSOR})

endif()

list(

APPEND

CUPTI_CHECK_LIBRARY_DIRS

${CUPTI_ROOT}

${CUPTI_ROOT}/lib64

${CUPTI_ROOT}/lib

${CUPTI_ROOT}/lib/${TARGET_ARCH}-linux-gnu

$ENV{CUPTI_ROOT}

$ENV{CUPTI_ROOT}/lib64

$ENV{CUPTI_ROOT}/lib

/usr/lib

${CUDA_TOOLKIT_ROOT_DIR}/targets/x86_64-linux/lib64

${CUDA_TOOLKIT_ROOT_DIR}/extras/CUPTI/lib64)

find_library(

CUPTI_LIBRARY

NAMES libcupti.so libcupti.dylib # libcupti_static.a

PATHS ${CUPTI_CHECK_LIBRARY_DIRS} ${CUPTI_INCLUDE_DIR} ${__libpath_hist}

NO_DEFAULT_PATH

DOC "Path to cuPTI library.")

get_filename_component(CUPTI_LIBRARY_PATH ${CUPTI_LIBRARY} DIRECTORY)

if(CUPTI_INCLUDE_DIR AND CUPTI_LIBRARY)

set(CUPTI_FOUND ON)

if(WITH_ROCM)

include_directories(${ROCM_PATH}/cuda/include)

add_definitions(-D__CUDA_HIP_PLATFORM_AMD__)

endif()

else()

set(CUPTI_FOUND OFF)

endif()