Merge pull request AdaptiveCpp#1293 from AdaptiveCpp/rocm6-platform-m…

…acros HIP: Update platform macros for ROCm 6
juhigupta0 · Dec 19, 2023 · 152054a · 152054a
2 parents b05729d + 52344d7
commit 152054a
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/include/hipSYCL/runtime/hip/hip_target.hpp b/include/hipSYCL/runtime/hip/hip_target.hpp
@@ -37,11 +37,17 @@
 #endif
 
 #ifdef HIPSYCL_RT_HIP_TARGET_CUDA
-#define __HIP_PLATFORM_NVCC__
+#define __HIP_PLATFORM_NVCC__ // Only needed for ROCm <6.0
+#define __HIP_PLATFORM_NVIDIA__
 #include <hip/hip_runtime.h>
 #elif defined(HIPSYCL_RT_HIP_TARGET_ROCM)
-#ifndef __HIP_PLATFORM_HCC__
+#ifndef __HIP_PLATFORM_HCC__ // Only needed for ROCm <6.0
 #define __HIP_PLATFORM_HCC__
+#endif
+#ifndef __HIP_PLATFORM_AMD__
+#define __HIP_PLATFORM_AMD__
+#endif
+#ifndef __HIP_ROCclr__
 #define __HIP_ROCclr__ (1)
 #endif
 #include <hip/hip_runtime.h>