Disable AVX512 support for NVHPC

nliber · Jun 15, 2023 · 352f4c3 · 352f4c3
1 parent 3163aef
commit 352f4c3
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/simd/src/Kokkos_SIMD.hpp b/simd/src/Kokkos_SIMD.hpp
@@ -25,7 +25,8 @@
 #include <Kokkos_SIMD_AVX2.hpp>
 #endif
 
-#ifdef KOKKOS_ARCH_AVX512XEON
+// FIXME_NVHPC nvc++ doesn't seem to support AVX512.
+#if !defined(KOKKOS_COMPILER_NVHPC) && defined(KOKKOS_ARCH_AVX512XEON)
 #include <Kokkos_SIMD_AVX512.hpp>
 #endif
 
@@ -40,7 +41,8 @@ namespace simd_abi {
 
 namespace Impl {
 
-#if defined(KOKKOS_ARCH_AVX512XEON)
+// FIXME_NVHPC nvc++ doesn't seem to support AVX512.
+#if !defined(KOKKOS_COMPILER_NVHPC) && defined(KOKKOS_ARCH_AVX512XEON)
 using host_native = avx512_fixed_size<8>;
 #elif defined(KOKKOS_ARCH_AVX2)
 using host_native  = avx2_fixed_size<4>;
@@ -139,7 +141,8 @@ class abi_set {};
 template <typename... Ts>
 class data_types {};
 
-#if defined(KOKKOS_ARCH_AVX512XEON)
+// FIXME_NVHPC nvc++ doesn't seem to support AVX512.
+#if !defined(KOKKOS_COMPILER_NVHPC) && defined(KOKKOS_ARCH_AVX512XEON)
 using host_abi_set  = abi_set<simd_abi::scalar, simd_abi::avx512_fixed_size<8>>;
 using data_type_set = data_types<std::int32_t, std::uint32_t, std::int64_t,
                                  std::uint64_t, double>;