Move GPU 3D FFT code to fft module

[alexxy/gromacs.git] / src / gromacs / fft / CMakeLists.txt

diff --git a/src/gromacs/fft/CMakeLists.txt b/src/gromacs/fft/CMakeLists.txt
index bbcd6972a8a2505eca7473d1929a71d1f7e5a966..70dd951e6f92e635173342fa745f1807ea4d587d 100644 (file)
--- a/src/gromacs/fft/CMakeLists.txt
+++ b/src/gromacs/fft/CMakeLists.txt
@@ -1,7 +1,7 @@
 #
 # This file is part of the GROMACS molecular simulation package.
 #
-# Copyright (c) 2013,2014,2015,2018,2019,2020, by the GROMACS development team, led by
+# Copyright (c) 2013,2014,2015,2018,2019,2020,2021, by the GROMACS development team, led by
 # Mark Abraham, David van der Spoel, Berk Hess, and Erik Lindahl,
 # and including many others, as listed in the AUTHORS file in the
 # top-level source directory and at http://www.gromacs.org.
@@ -54,6 +54,32 @@ if (GMX_FFT_MKL)
     gmx_add_libgromacs_sources(fft_mkl.cpp)
 endif()
 
+if (GMX_GPU_CUDA)
+    gmx_add_libgromacs_sources(
+        # CUDA-specific sources
+        gpu_3dfft.cu
+        )
+elseif (GMX_GPU_OPENCL)
+    gmx_add_libgromacs_sources(
+        # OpenCL-specific sources
+        gpu_3dfft_ocl.cpp
+        )
+elseif (GMX_GPU_SYCL)
+    # SYCL-TODO: proper implementation
+    gmx_add_libgromacs_sources(
+        # SYCL-specific sources
+        gpu_3dfft_sycl.cpp
+        )
+    _gmx_add_files_to_property(SYCL_SOURCES
+        gpu_3dfft_sycl.cpp
+        )
+else()
+    gmx_add_libgromacs_sources(
+        # Stub sources for CPU-only build
+        gpu_3dfft.cpp
+        )
+endif()
+
 # Source files have the following private module dependencies.
 target_link_libraries(fft PRIVATE
                       #                      gmxlib