Add roctracer support

wdmapp · Jan 3, 2024 · d80c357 · d80c357
1 parent 579fb24
commit d80c357
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -200,6 +200,8 @@ if ("hip" IN_LIST GTENSOR_BUILD_DEVICES)
     target_compile_definitions(gtensor_hip INTERFACE GTENSOR_USE_THRUST)
   endif()
 
+  target_include_directories(gtensor_hip INTERFACE ${ROCM_PATH}/include/roctracer)
+  target_link_libraries(gtensor_hip INTERFACE ${ROCM_PATH}/lib/libroctracer64.so)
   target_link_libraries(gtensor_hip INTERFACE roc::rocthrust)
 
   if (HIP_GCC_TOOLCHAIN)

diff --git a/src/fortran/gpu_api.cxx b/src/fortran/gpu_api.cxx
@@ -28,6 +28,7 @@
 #include "cuda_runtime_api.h"
 #elif defined(GTENSOR_DEVICE_HIP)
 #include "hip/hip_runtime.h"
+#include "roctracer_ext.h"
 #endif
 
 #ifdef GTENSOR_DEVICE_SYCL
@@ -57,13 +58,17 @@ extern "C" void gpuProfilerStart()
 {
 #ifdef GTENSOR_DEVICE_CUDA
   gtGpuCheck(cudaProfilerStart());
+#elif defined(GTENSOR_DEVICE_HIP)
+  roctracer_start();
 #endif
 }
 
 extern "C" void gpuProfilerStop()
 {
 #ifdef GTENSOR_DEVICE_CUDA
   gtGpuCheck(cudaProfilerStop());
+#elif defined(GTENSOR_DEVICE_HIP)
+  roctracer_stop();
 #endif
 }