未验证 提交 14aed740 编写于 作者: R ronnywang 提交者: GitHub

[ROCM] fix profiler compilation error on dtk23.04 (#56079)

上级 12823f2b
...@@ -4,7 +4,7 @@ endif() ...@@ -4,7 +4,7 @@ endif()
if(WITH_ROCM) if(WITH_ROCM)
set(CUPTI_ROOT set(CUPTI_ROOT
"${ROCM_PATH}/CUPTI" "${ROCM_PATH}/cuda/extras/CUPTI"
CACHE PATH "CUPTI ROOT") CACHE PATH "CUPTI ROOT")
else() else()
set(CUPTI_ROOT set(CUPTI_ROOT
...@@ -52,6 +52,10 @@ find_library( ...@@ -52,6 +52,10 @@ find_library(
get_filename_component(CUPTI_LIBRARY_PATH ${CUPTI_LIBRARY} DIRECTORY) get_filename_component(CUPTI_LIBRARY_PATH ${CUPTI_LIBRARY} DIRECTORY)
if(CUPTI_INCLUDE_DIR AND CUPTI_LIBRARY) if(CUPTI_INCLUDE_DIR AND CUPTI_LIBRARY)
set(CUPTI_FOUND ON) set(CUPTI_FOUND ON)
if(WITH_ROCM)
include_directories(${ROCM_PATH}/cuda/include)
add_definitions(-D__CUDA_HIP_PLATFORM_AMD__)
endif()
else() else()
set(CUPTI_FOUND OFF) set(CUPTI_FOUND OFF)
endif() endif()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册