cmake_minimum_required (VERSION 3.5)
find_package( OpenCV  REQUIRED )

message(STATUS "oPENCV Library status:")
message(STATUS ">version:${OpenCV_VERSION}")
message(STATUS "Include:${OpenCV_INCLUDE_DIRS}")


set(CMAKE_CXX_STANDARD 17)

set(CMAKE_CXX_FLAGS "-Wno-error=deprecated-declarations -Wno-deprecated-declarations")

find_package(CUDA REQUIRED)
message(STATUS "cuda version: " ${CUDA_VERSION_STRING})
message(STATUS "cuda CUDA_INCLUDE_DIRS: " ${CUDA_INCLUDE_DIRS})
include_directories(${CUDA_INCLUDE_DIRS})

include_directories(
${CMAKE_CURRENT_SOURCE_DIR}/include
${PROJECT_SOURCE_DIR}/AIEngineModule/include
${PROJECT_SOURCE_DIR}/AIEngineModule/include_base
)

link_directories(
/usr/local/lib/
)
set(CMAKE_CUDA_ARCHITECTURES 86)
set(CMAKE_CUDA_COMPILER "/usr/local/cuda/bin/nvcc") 
enable_language(CUDA)
file(GLOB SRC_LISTS 
    ${CMAKE_CURRENT_SOURCE_DIR}/*.cpp  
    ${PROJECT_SOURCE_DIR}/AIEngineModule/src/*.cpp
    ${PROJECT_SOURCE_DIR}/AIEngineModule/src/*.c
    ${PROJECT_SOURCE_DIR}/AIEngineModule/src/*.cu  
)

add_executable("ai_test" ${SRC_LISTS})

target_link_libraries("ai_test"
	pthread
	z
	nvinfer
    ${CUDA_LIBRARIES}
	${OpenCV_LIBS}
)
