[UR][HIP] Detect COMGR version in CMake

Signed-off-by: Lukas Sommer <[email protected]>

Author: sommerlukas

source/adapters/hip/CMakeLists.txt

@@ -101,6 +101,21 @@ if("${UR_HIP_PLATFORM}" STREQUAL "AMD")
     )
 
     if(UR_ENABLE_COMGR)
+        set(UR_COMGR_VERSION5_HEADER "${UR_HIP_INCLUDE_DIR}/amd_comgr/amd_comgr.h")
+        set(UR_COMGR_VERSION4_HEADER "${UR_HIP_INCLUDE_DIR}/amd_comgr.h")
+        # The COMGR header changed location between ROCm versions 4 and 5.
+        # Check for existence in the version 5 location or fallback to version 4
+        if(NOT EXISTS "${UR_COMGR_VERSION5_HEADER}")
+            if(NOT EXISTS "${UR_COMGR_VERSION4_HEADER}")
+                message(FATAL_ERROR "Could not find AMD COMGR header at "
+                                    "${UR_COMGR_VERSION5_HEADER} or"
+                                    "${UR_COMGR_VERSION4_HEADER}, "
+                                    "check ROCm installation")
+            else()
+                target_compile_definitions(${TARGET_NAME} PRIVATE UR_COMGR_VERSION4_INCLUDE)
+            endif()
+        endif()
+
         add_library(amd_comgr SHARED IMPORTED GLOBAL)
         set_target_properties(
         amd_comgr PROPERTIES

source/adapters/hip/common.hpp

@@ -10,11 +10,10 @@
 #pragma once
 
 #ifdef SYCL_ENABLE_KERNEL_FUSION
-#include <rocm/rocm_version.h>
-#if (ROCM_VERSION_MAJOR >= 5)
-#include <amd_comgr/amd_comgr.h>
-#else
+#ifdef UR_COMGR_VERSION4_INCLUDE
 #include <amd_comgr.h>
+#else
+#include <amd_comgr/amd_comgr.h>
 #endif
 #endif
 #include <hip/hip_runtime.h>

source/adapters/hip/program.cpp

@@ -11,11 +11,10 @@
 #include "program.hpp"
 
 #ifdef SYCL_ENABLE_KERNEL_FUSION
-#include <rocm/rocm_version.h>
-#if (ROCM_VERSION_MAJOR >= 5)
-#include <amd_comgr/amd_comgr.h>
-#else
+#ifdef UR_COMGR_VERSION4_INCLUDE
 #include <amd_comgr.h>
+#else
+#include <amd_comgr/amd_comgr.h>
 #endif
 namespace {
 template <typename ReleaseType, ReleaseType Release, typename T>