Added compatibility for Opencv 3 in cuFFT.

diff --git a/src/fft_cufft.h b/src/fft_cufft.h
index 15ee2a48949e7678921be3ffc7694bb6a86fe88e..3d22aa16224fdf6c2c8765b7c734a674e8faf93e 100644 (file)
--- a/src/fft_cufft.h
+++ b/src/fft_cufft.h
@@ -4,6 +4,14 @@
 
 #include "fft.h"
 
+#if CV_MAJOR_VERSION == 2
+  #include <opencv2/gpu/gpu.hpp>
+  #define CUDA cv::gpu
+#else
+  #include "opencv2/opencv.hpp"
+  #define CUDA cv::cuda
+#endif
+
 #include <cufft.h>
 #include <cuda_runtime.h>