mpi_fft_engine

fftransform_cufft.hh
View Options

			/**
			* @file
			* @section LICENSE
			*
			* Copyright (©) 2016-2020 EPFL (École Polytechnique Fédérale de Lausanne),
			* Laboratory (LSMS - Laboratoire de Simulation en Mécanique des Solides)
			*
			* This program is free software: you can redistribute it and/or modify
			* it under the terms of the GNU Affero General Public License as published
			* by the Free Software Foundation, either version 3 of the License, or
			* (at your option) any later version.
			*
			* This program is distributed in the hope that it will be useful,
			* but WITHOUT ANY WARRANTY; without even the implied warranty of
			* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
			* GNU Affero General Public License for more details.
			*
			* You should have received a copy of the GNU Affero General Public License
			* along with this program. If not, see <https://www.gnu.org/licenses/>.
			*
			*/

			/* -------------------------------------------------------------------------- */

			#ifndef __FFTRANSFORM_CUFFT_HH__
			#define __FFTRANSFORM_CUFFT_HH__

			#include "fftransform.hh"
			#include "grid.hh"
			#include <cufft.h>

			namespace tamaas {

			template <typename T, UInt dim>
			class FFTransformCUFFT : public FFTransform<T, dim> {
			public:
			/// Constructor
			FFTransformCUFFT(Grid<T, dim>& real, GridHermitian<T, dim>& spectral);

			/// Destructor
			virtual ~FFTransformCUFFT();

			public:
			/// Perform FFT
			void forwardTransform() override;

			/// Perform IFFT
			void backwardTransform() override;

			private:
			cufftHandle forward_plan;
			cufftHandle backward_plan;
			};

			} // namespace tamaas

			#endif // __FFTRANSFORM_CUFFT_HH__