Added safety check for image dimensions in cuFFT version.

diff --git a/src/fft_cufft.cpp b/src/fft_cufft.cpp
index c94d61f14cdd6c3a790a38a7e836042447a60b2f..4e0142625b7ba0daebc20ca51d8878d280572d42 100644 (file)
--- a/src/fft_cufft.cpp
+++ b/src/fft_cufft.cpp
@@ -84,6 +84,10 @@ void cuFFT::init(unsigned width, unsigned height, unsigned num_of_feats, unsigne
 
     std::cout << "FFT: cuFFT" << std::endl;
     
+    if(m_height*(m_width/2+1) > 1024){
+        std::cerr << "Image dimension after forward FFT are too big for CUDA kernels." << std::endl;
+        std::exit(EXIT_FAILURE);
+    }
 
     for (unsigned i = 0; i < m_num_of_streams; i++) cudaStreamCreate(&streams[i]);