Update Kokkos_SIMD.hpp

Remove the check for nvc here and add it to cmake.
crtrott · Jul 25, 2024 · be4c315 · be4c315
1 parent b472df8
commit be4c315
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/simd/src/Kokkos_SIMD.hpp b/simd/src/Kokkos_SIMD.hpp
@@ -88,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) && !defined(__NVCC__)
+#elif defined(KOKKOS_ARCH_ARM_NEON) 
 using host_native  = neon_fixed_size<2>;
 #else
 using host_native   = scalar;
@@ -193,7 +193,7 @@ using host_abi_set = abi_set<simd_abi::scalar, simd_abi::avx2_fixed_size<4>,
                              simd_abi::avx2_fixed_size<8>>;
 using data_type_set =
     data_types<std::int32_t, std::int64_t, std::uint64_t, double, float>;
-#elif defined(KOKKOS_ARCH_ARM_NEON)  && !defined(__NVCC__)
+#elif defined(KOKKOS_ARCH_ARM_NEON)
 using host_abi_set = abi_set<simd_abi::scalar, simd_abi::neon_fixed_size<2>,
                              simd_abi::neon_fixed_size<4>>;
 using data_type_set =