Update HeterogeneousCore/ROCmServices/plugins/ROCmService.cc

Co-authored-by: Andrea Bocci <[email protected]>
cms-sw · Oct 20, 2024 · 8021b14 · 8021b14
1 parent 44d79e8
commit 8021b14
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/HeterogeneousCore/ROCmServices/plugins/ROCmService.cc b/HeterogeneousCore/ROCmServices/plugins/ROCmService.cc
@@ -6,7 +6,12 @@
 #include <vector>
 
 #include <hip/hip_runtime.h>
+#if HIP_VERSION_MAJOR >= 6
+// the location of rocm_version.h changed in HIP/ROCm 6.0
+#include <rocm-core/rocm_version.h>
+#else
 #include <rocm_version.h>
+#endif  // HIP_VERSION_MAJOR
 #include <rocm_smi/rocm_smi.h>
 
 #include "FWCore/MessageLogger/interface/MessageLogger.h"