diff --git a/projects/rdc/rdc_libs/rdc/CMakeLists.txt b/projects/rdc/rdc_libs/rdc/CMakeLists.txt index b8a338ab35..45e401f338 100644 --- a/projects/rdc/rdc_libs/rdc/CMakeLists.txt +++ b/projects/rdc/rdc_libs/rdc/CMakeLists.txt @@ -83,3 +83,6 @@ if("${CMAKE_BUILD_TYPE}" STREQUAL Release) TARGET ${RDC_LIB} POST_BUILD COMMAND ${CMAKE_STRIP} ${RDC_LIB_COMPONENT}.so) endif() + +# pass ROCM_DIR to compiler +target_compile_definitions(${RDC_LIB} PRIVATE ROCM_DIR="${ROCM_DIR}") \ No newline at end of file diff --git a/projects/rdc/rdc_libs/rdc/src/RdcRocpLib.cc b/projects/rdc/rdc_libs/rdc/src/RdcRocpLib.cc index dc43323408..c4cbff4d06 100644 --- a/projects/rdc/rdc_libs/rdc/src/RdcRocpLib.cc +++ b/projects/rdc/rdc_libs/rdc/src/RdcRocpLib.cc @@ -154,7 +154,7 @@ rdc_status_t RdcRocpLib::rdc_telemetry_fields_unwatch(rdc_gpu_field_t* fields, std::string RdcRocpLib::get_rocm_path() { // set default rocm path in case lookup fails - std::string rocm_path("/opt/rocm"); + std::string rocm_path(ROCM_DIR); const char* rocm_path_env = getenv("ROCM_PATH"); if (rocm_path_env != nullptr) { rocm_path = rocm_path_env;