Add check to ensure complex alignment (#228)

* Add check to ensure complex alignment * fix kokkos_check usage --------- Co-authored-by: Yuuichi Asahi <y.asahi@nr.titech.ac.jp>
kokkos · Feb 20, 2025 · 02692b1 · 02692b1
1 parent 14a90b6
commit 02692b1
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -38,6 +38,8 @@ if (NOT KokkosFFT_ENABLE_INTERNAL_KOKKOS)
     # First check, Kokkos is added as subdirectory or not
     if(NOT TARGET Kokkos::kokkos)
         find_package(Kokkos ${KOKKOS_REQUIRED_VERSION} REQUIRED)
+        # Check the alignment of complex numbers
+        kokkos_check(OPTIONS COMPLEX_ALIGN)
     endif()
 else ()
     add_subdirectory(tpls/kokkos)

diff --git a/fft/src/KokkosFFT_default_types.hpp b/fft/src/KokkosFFT_default_types.hpp
@@ -7,6 +7,12 @@
 
 #include <Kokkos_Core.hpp>
 
+#if !defined(KOKKOS_ENABLE_COMPLEX_ALIGN)
+static_assert(false,
+              "KokkosFFT requires option -DKokkos_ENABLE_COMPLEX_ALIGN=ON to "
+              "build Kokkos");
+#endif
+
 #if defined(KOKKOS_ENABLE_CUDA)
 #include "KokkosFFT_Cuda_types.hpp"
 #elif defined(KOKKOS_ENABLE_HIP)