Skip to content

Commit

Permalink
simplify fftw plan wrapper
Browse files Browse the repository at this point in the history
  • Loading branch information
Yuuichi Asahi committed Dec 19, 2024
1 parent 27501ac commit 4bde4d2
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions fft/src/KokkosFFT_FFTW_Types.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,6 @@ struct ScopedFFTWPlan {
std::conditional_t<std::is_same_v<floating_point_type, float>, fftwf_plan,
fftw_plan>;
plan_type m_plan;
bool m_is_created = false;

public:
template <typename InScalarType, typename OutScalarType>
Expand Down Expand Up @@ -102,15 +101,14 @@ struct ScopedFFTWPlan {
m_plan = fftw_plan_many_dft(rank, n, howmany, in, inembed, istride, idist,
out, onembed, ostride, odist, sign, flags);
}
m_is_created = true;
}

~ScopedFFTWPlan() noexcept {
cleanup_threads();
if constexpr (std::is_same_v<plan_type, fftwf_plan>) {
if (m_is_created) fftwf_destroy_plan(m_plan);
fftwf_destroy_plan(m_plan);
} else {
if (m_is_created) fftw_destroy_plan(m_plan);
fftw_destroy_plan(m_plan);
}
}

Expand Down

0 comments on commit 4bde4d2

Please sign in to comment.