fft_cuda.hpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2025 VTT Technical Research Centre of Finland Ltd
// SPDX-License-Identifier: AGPL-3.0-or-later
 
#pragma once
 
#include "openpfc/core/decomposition.hpp"
#include "openpfc/fft.hpp"
 
#include <heffte.h>
#include <mpi.h>
 
namespace pfc {
namespace fft {
 
#if defined(OpenPFC_ENABLE_CUDA)
// GPU FFT type alias
using FFT_CUDA = FFT_Impl<heffte::backend::cufft>;
 
FFT_CUDA create_cuda(const Decomposition &decomposition, int rank_id);
 
FFT_CUDA create_cuda(const Decomposition &decomposition);
#endif // OpenPFC_ENABLE_CUDA
 
} // namespace fft
} // namespace pfc