Cheetah - SKA - PSS - Prototype Time Domain Search Pipeline
FftPlan.cu
1 #include "cheetah/fft/cuda/FftPlan.cuh"
2 
3 namespace ska {
4 namespace cheetah {
5 namespace fft {
6 namespace cuda {
7 
8 template <typename T>
9 cufftHandle const& FftPlan::plan(FftType fft_type, std::size_t size, std::size_t batch)
10 {
11  cufftType cufft_type = convert_to_cufft_type<T>(fft_type);
12  if (!valid(cufft_type,size,batch))
13  {
14  destroy_plan();
15  _size = size;
16  _cufft_type = cufft_type;
17  _batch = batch;
18  CUFFT_ERROR_CHECK(cufftPlan1d(&_plan, (int) _size, _cufft_type, (int) _batch));
19  }
20  return _plan;
21 }
22 
23 } // namespace cuda
24 } // namespace fft
25 } // namespace cheetah
26 } // namespace ska
cufftHandle const & plan(FftType fft_type, std::size_t size, std::size_t batch)
Get (or create) a cufft plan.
Definition: FftPlan.cu:9
Some limits and constants for FLDO.
Definition: Brdz.h:35