HIP: Restrict AVX2 workaround to ROCm 5.6 and 5.7

tcclevenger · Sep 21, 2023 · b813f2b · b813f2b
1 parent e1c8266
commit b813f2b
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/simd/src/Kokkos_SIMD_AVX2.hpp b/simd/src/Kokkos_SIMD_AVX2.hpp
@@ -25,10 +25,9 @@
 
 #include <immintrin.h>
 
-// FIXME_HIP ROCm 5.6 can't compile with the intrinsic used here.
-#if defined(__HIPCC__) &&       \
-    ((HIP_VERSION_MAJOR > 5) || \
-     ((HIP_VERSION_MAJOR == 5) && (HIP_VERSION_MINOR >= 6)))
+// FIXME_HIP ROCm 5.6 and 5.7 can't compile with the intrinsic used here.
+#if defined(__HIPCC__) && (HIP_VERSION_MAJOR == 5) && \
+    ((HIP_VERSION_MINOR == 6) || (HIP_VERSION_MINOR == 7))
 #define KOKKOS_IMPL_WORKAROUND_ROCM_AVX2_ISSUE
 #endif