Update kokkos_arch.cmake

Disable NEON on ARM if CUDA is enabled since nvc can not compile that.
crtrott · Jul 25, 2024 · 089f2a8 · 089f2a8
1 parent be4c315
commit 089f2a8
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/cmake/kokkos_arch.cmake b/cmake/kokkos_arch.cmake
@@ -550,6 +550,11 @@ IF (KOKKOS_CXX_HOST_COMPILER_ID STREQUAL NVHPC)
   SET(KOKKOS_ARCH_AVX512XEON OFF)
 ENDIF()
 
+# FIXME_NVHPC nvc++ doesn't seem to support Arm Neon.
+IF(KOKKOS_CXX_COMPILER_ID STREQUAL NVIDIA)
+  SET(KOKKOS_ARCH_ARM_NEON OFF)
+ENDIF()
+
 IF (NOT KOKKOS_COMPILE_LANGUAGE STREQUAL CUDA)
   IF (KOKKOS_ENABLE_CUDA_RELOCATABLE_DEVICE_CODE)
       COMPILER_SPECIFIC_FLAGS(