Fix hipSYCL build with CUDA target

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

diff --git a/src/gromacs/fft/CMakeLists.txt b/src/gromacs/fft/CMakeLists.txt
index 70dd951e6f92e635173342fa745f1807ea4d587d..16a5a4b7056c3454da1856faf9a737bf5405fe22 100644 (file)
--- a/src/gromacs/fft/CMakeLists.txt
+++ b/src/gromacs/fft/CMakeLists.txt
@@ -40,6 +40,7 @@ gmx_add_libgromacs_sources(
      fft.cpp
      fft5d.cpp
      parallel_3dfft.cpp
+     gpu_3dfft.cpp
      )
 
 if (GMX_FFT_FFTPACK)
@@ -53,11 +54,19 @@ endif()
 if (GMX_FFT_MKL)
     gmx_add_libgromacs_sources(fft_mkl.cpp)
 endif()
-
+if(Heffte_FOUND)
+    gmx_add_libgromacs_sources(
+        gpu_3dfft_heffte.cpp
+        )
+endif()
 if (GMX_GPU_CUDA)
     gmx_add_libgromacs_sources(
         # CUDA-specific sources
-        gpu_3dfft.cu
+        gpu_3dfft_cufft.cu
+        )
+    _gmx_add_files_to_property(CUDA_SOURCES
+        # Must add these files so they can include cuda_runtime.h
+        gpu_3dfft.cpp
         )
 elseif (GMX_GPU_OPENCL)
     gmx_add_libgromacs_sources(
@@ -65,13 +74,28 @@ elseif (GMX_GPU_OPENCL)
         gpu_3dfft_ocl.cpp
         )
 elseif (GMX_GPU_SYCL)
-    # SYCL-TODO: proper implementation
+    if (NOT GMX_SYCL_HIPSYCL AND GMX_FFT_MKL)
+            gmx_add_libgromacs_sources(
+            gpu_3dfft_sycl_mkl.cpp
+            )
+        _gmx_add_files_to_property(SYCL_SOURCES
+            gpu_3dfft_sycl_mkl.cpp
+            )
+    endif()
+    if (GMX_SYCL_HIPSYCL AND GMX_HIPSYCL_HAVE_HIP_TARGET)
+            gmx_add_libgromacs_sources(
+            gpu_3dfft_sycl_rocfft.cpp
+            )
+        _gmx_add_files_to_property(SYCL_SOURCES
+            gpu_3dfft_sycl_rocfft.cpp
+            )
+    endif()
     gmx_add_libgromacs_sources(
-        # SYCL-specific sources
         gpu_3dfft_sycl.cpp
         )
     _gmx_add_files_to_property(SYCL_SOURCES
         gpu_3dfft_sycl.cpp
+        gpu_3dfft.cpp
         )
 else()
     gmx_add_libgromacs_sources(