From 14aed740e82bc575cad945bbe6e568e16b9c2219 Mon Sep 17 00:00:00 2001
From: ronnywang <ronny1996@163.com>
Date: Wed, 9 Aug 2023 09:56:24 +0800
Subject: [PATCH] [ROCM] fix profiler compilation error on dtk23.04 (#56079)

---
 cmake/cupti.cmake | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/cmake/cupti.cmake b/cmake/cupti.cmake
index 54905d5842f..eb7ad44af23 100644
--- a/cmake/cupti.cmake
+++ b/cmake/cupti.cmake
@@ -4,7 +4,7 @@ endif()
 
 if(WITH_ROCM)
   set(CUPTI_ROOT
-      "${ROCM_PATH}/CUPTI"
+      "${ROCM_PATH}/cuda/extras/CUPTI"
       CACHE PATH "CUPTI ROOT")
 else()
   set(CUPTI_ROOT
@@ -52,6 +52,10 @@ find_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()
-- 
GitLab