Work done so far on CUDA streams

[hercules2020/kcf.git] / src / CMakeLists.txt

diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index 80350481b52e79c4901f8bd3268a2954120b9b5b..da31a5ead2e2ddd22ae4960c10098790bdf290c5 100644 (file)
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -61,9 +61,12 @@ IF((FFT STREQUAL "cuFFT") AND (ASYNC))
   message(SEND_ERROR "cuFFT version does not support ASYNC and OpenMP only if used with big batch mode.")
 ENDIF()
 
-IF(ASYNC AND NOT OPENMP)
+IF(ASYNC)
   add_definitions(-DASYNC)
   MESSAGE(STATUS "ASYNC")
+ELSEIF(OPENMP)
+    add_definitions(-DOPENMP)
+    MESSAGE(STATUS "OPENMP")
 ENDIF() #ASYNC
 
 FIND_PACKAGE( OpenCV REQUIRED )
@@ -79,7 +82,7 @@ IF(use_cuda)
   set(CUDA_SEPARABLE_COMPILATION ON)
   set(CUDA_PROPAGATE_HOST_FLAGS OFF)
   set(CUDA_HOST_COMPILER /usr/bin/g++)
-  list( APPEND CUDA_NVCC_FLAGS -O3 --gpu-architecture sm_62 -std=c++11 -default-stream per-thread)
+  list( APPEND CUDA_NVCC_FLAGS -O3 --gpu-architecture sm_62 -std=c++11)
   find_cuda_helper_libs(cufftw)
   IF(FFT STREQUAL "cuFFT")
   add_subdirectory(cuda)