Disable NEON simd when compiling with NVCC

kokkos · May 23, 2024 · 24d9381 · 24d9381
1 parent 34547c2
commit 24d9381
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/simd/src/Kokkos_SIMD.hpp b/simd/src/Kokkos_SIMD.hpp
@@ -62,7 +62,8 @@
 #include <Kokkos_SIMD_AVX512.hpp>
 #endif
 
-#if defined(KOKKOS_ARCH_ARM_NEON)
+//FIXME: If and when NVCC supports NEON intrinsics reenable
+#if defined(KOKKOS_ARCH_ARM_NEON) && !defined(__NVCC__)
 #if !defined(__ARM_NEON)
 #error "__ARM_NEON must be definded for KOKKOS_ARCH_ARM_NEON"
 #endif
@@ -87,7 +88,7 @@ namespace Impl {
 using host_native = avx512_fixed_size<8>;
 #elif defined(KOKKOS_ARCH_AVX2)
 using host_native  = avx2_fixed_size<4>;
-#elif defined(KOKKOS_ARCH_ARM_NEON)
+#elif defined(KOKKOS_ARCH_ARM_NEON) && !defined(__NVCC__)
 using host_native  = neon_fixed_size<2>;
 #else
 using host_native   = scalar;
@@ -190,7 +191,7 @@ using data_type_set = data_types<std::int32_t, std::uint32_t, std::int64_t,
 using host_abi_set = abi_set<simd_abi::scalar, simd_abi::avx2_fixed_size<4>>;
 using data_type_set =
     data_types<std::int32_t, std::int64_t, std::uint64_t, double, float>;
-#elif defined(KOKKOS_ARCH_ARM_NEON)
+#elif defined(KOKKOS_ARCH_ARM_NEON) && !defined(__NVCC__)
 using host_abi_set = abi_set<simd_abi::scalar, simd_abi::neon_fixed_size<2>>;
 using data_type_set =
     data_types<std::int32_t, std::int64_t, std::uint64_t, double, float>;