cmake_minimum_required(VERSION 3.16)

set(ENV{CUDACXX} /usr/local/cuda/bin/nvcc)

project(tf-gpu-feed LANGUAGES CXX CUDA)

cmake_policy(SET CMP0074 OLD)

set(CMAKE_CXX_STANDARD 11)

find_package(CUDA 10.0 REQUIRED)

set(SOURCE_FILES
    main.cpp
    dynlink_nvcuvid.cpp
    array.cu
    )

set(TENSORFLOW_LIBS libtensorflow_cc.so libtensorflow_framework.so)

link_directories("/usr/local/tensorflow/lib")

add_executable(tf-gpu-feed ${SOURCE_FILES})
set_target_properties(tf-gpu-feed PROPERTIES CUDA_SEPARABLE_COMPILATION ON)

include_directories(${CUDA_INCLUDE_DIRS})

# TensorFlow headers
include_directories("/usr/local/tensorflow/include/tensorflow/")
include_directories("/usr/local/tensorflow/include/third-party/")
include_directories("/usr/local/tensorflow/include/")

target_link_libraries(tf-gpu-feed
    ${CUDA_LIBRARIES}
    cuda
    cublas
    nvinfer
    nvToolsExt
    nvparsers
    nvinfer_plugin
    nvonnxparser
    ${CMAKE_DL_LIBS}
    ${TENSORFLOW_LIBS})