Move GPU 3D FFT code to fft module

[alexxy/gromacs.git] / src / gromacs / ewald / pme_gpu_types_host_impl.h

diff --git a/src/gromacs/ewald/pme_gpu_types_host_impl.h b/src/gromacs/ewald/pme_gpu_types_host_impl.h
index 6b16ba5c16ca54120cf1a8367e1c638ff4d2f61d..39b195d3295a41590231f0f4bf447acce7bcc2c3 100644 (file)
--- a/src/gromacs/ewald/pme_gpu_types_host_impl.h
+++ b/src/gromacs/ewald/pme_gpu_types_host_impl.h
@@ -62,16 +62,18 @@
 #    include "gromacs/gpu_utils/gpuregiontimer_sycl.h"
 #endif
 
+#include "gromacs/fft/gpu_3dfft.h"
 #include "gromacs/timing/gpu_timing.h" // for gtPME_EVENT_COUNT
 
-#include "pme_gpu_3dfft.h"
-
 #ifndef NUMFEPSTATES
 //! Number of FEP states.
 #    define NUMFEPSTATES 2
 #endif
 
-class GpuParallel3dFft;
+namespace gmx
+{
+class Gpu3dFft;
+} // namespace gmx
 
 /*! \internal \brief
  * The main PME CUDA/OpenCL-specific host data structure, included in the PME GPU structure by the archSpecific pointer.
@@ -116,7 +118,7 @@ struct PmeGpuSpecific
     bool useTiming = false;
 
     //! Vector of FFT setups
-    std::vector<std::unique_ptr<GpuParallel3dFft>> fftSetup;
+    std::vector<std::unique_ptr<gmx::Gpu3dFft>> fftSetup;
 
     //! All the timers one might use
     gmx::EnumerationArray<PmeStage, GpuRegionTimer> timingEvents;