Skip to content

Commit

Permalink
OpenMPTarget: Disable check for SIMD compiler macros
Browse files Browse the repository at this point in the history
Co-authored-by: Rahulkumar Gayatri <rgayatri@lbl.gov>
  • Loading branch information
masterleinad and Rahulkumar Gayatri committed Sep 18, 2023
1 parent 9081d36 commit 7284cd2
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions simd/src/Kokkos_SIMD.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,20 @@

#include <Kokkos_Macros.hpp>

// FIXME_OPENMPTARGET The device pass disables all compiler macros checked
#ifdef KOKKOS_ENABLE_OPENMPTARGET
#if defined(KOKKOS_ARCH_AVX2)
#include <Kokkos_SIMD_AVX2.hpp>
#endif

#if defined(KOKKOS_ARCH_AVX512XEON)
#include <Kokkos_SIMD_AVX512.hpp>
#endif

#ifdef __ARM_NEON
#include <Kokkos_SIMD_NEON.hpp>
#endif
#else // KOKKOS_ENABLE_OPENMPTARGET
#if defined(KOKKOS_ARCH_AVX) && !defined(__AVX__)
#error "__AVX__ must be defined for KOKKOS_ARCH_AVX"
#endif
Expand All @@ -51,6 +65,7 @@
#ifdef __ARM_NEON
#include <Kokkos_SIMD_NEON.hpp>
#endif
#endif

#if defined(KOKKOS_COMPILER_NVCC) && KOKKOS_COMPILER_NVCC < 1130
#pragma GCC diagnostic pop
Expand Down

0 comments on commit 7284cd2

Please sign in to comment.