00001 00002 #ifndef eman__cuda_emfft_h__ 00003 #define eman__cuda_emfft_h__ 1 00004 00005 void init_cuda_fft_hh_plan_cache(); 00006 void cleanup_cuda_fft_hh_plan_cache(); 00007 00008 int cuda_hh_fft_real_to_complex_1d(float *real_data, float *complex_data, int n, int batch); 00009 int cuda_hh_fft_complex_to_real_1d(float *complex_data, float *real_data, int n, int batch); 00010 int cuda_hh_fft_real_to_complex_nd(float *real_data, float *complex_data, int nx, int ny, int nz); 00011 int cuda_hh_fft_complex_to_real_nd(float *complex_data, float *real_data, int nx, int ny, int nz); 00012 00013 void init_cuda_fft_dd_plan_cache(); 00014 void cleanup_cuda_fft_dd_plan_cache(); 00015 00016 int cuda_dd_fft_real_to_complex_1d(float *real_data, float *complex_data, int n, int batch); 00017 int cuda_dd_fft_complex_to_real_1d(float *complex_data, float *real_data, int n, int batch); 00018 int cuda_dd_fft_real_to_complex_nd(float *real_data, float *complex_data, int nx, int ny, int nz); 00019 int cuda_dd_fft_complex_to_real_nd(float *complex_data, float *real_data, int nx, int ny, int nz); 00020 00021 00022 #endif //eman__cuda_emfft_h__ 00023