CMAKE_MINIMUM_REQUIRED(VERSION 2.8)
PROJECT(myselu)
# CUDA package 安装在系统中直接用find_package, 否则用set(xx_DIR xxx) 引入include 和 lib
find_package(CUDA REQUIRED)
find_package(OpenCV REQUIRED)
#INCLUDE(FindCUDA)
# CUDA include directories
# file name

set(TENSORRT_DIR "/home/a/TensorRT-8.5.3.1/")
#set(CUDA_DIR "/usr/local/cuda/")
include_directories(
    ${TENSORRT_DIR}/include
    #${CUDA_DIR}/include
)
link_directories(
    ${TENSORRT_DIR}/lib
    #${CUDA_DIR}/lib
)
#sources file
FILE(GLOB SOURCES "*.cu" "*.cpp" "*.c" "*.h" "*.hpp")
# CUDA execu
CUDA_ADD_EXECUTABLE(myselu ${SOURCES})
# C++ list
LIST(APPEND CMAKE_CXX_FLAGS "-std=c++11 -O3 -ffast-math -Wall")
# CUDA list
LIST(APPEND CUDA_NVCC_FLAGS --compiler-options -fno-strict-aliasing -lineinfo -use_fast_math -Xptxas -dlcm=cg)
LIST(APPEND CUDA_NVCC_FLAGS -gencode arch=compute_20,code=sm_20)
LIST(APPEND CUDA_NVCC_FLAGS -gencode arch=compute_30,code=sm_30)
LIST(APPEND CUDA_NVCC_FLAGS -gencode arch=compute_35,code=sm_35)

# target
TARGET_LINK_LIBRARIES(myselu nvinfer nvonnxparser cuda ${OpenCV_LIBS})
add_definitions(-O2 -pthread)