Update HeterogeneousCore/ROCmServices/plugins/ROCmService.cc

Co-authored-by: Andrea Bocci <andrea.bocci@cern.ch>
cms-sw · Apr 24, 2024 · 6f7fa6a · 6f7fa6a
1 parent 96cd26b
commit 6f7fa6a
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"