From 56eacbff8658b7d71391e94d496da4c5dabb3d55 Mon Sep 17 00:00:00 2001 From: Shiroki Satsuki Date: Wed, 6 Jul 2022 15:36:38 +0800 Subject: [PATCH 1/2] feat: target_include_directories Add target_include_directories to CMakeLists.txt so that Trt.h can be automatically added to include path when `tinytrt` be in `target_link_libraries` --- CMakeLists.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 125d9f9..3ac9cfa 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -56,7 +56,6 @@ set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} ${CUDA_gencode_flags}") include_directories(third_party/spdlog/include) include_directories(third_party/pybind11/include) -include_directories(./) include_directories(./src) include_directories(./plugin) include_directories(${ZLIB_INCLUDE_DIRS}) @@ -84,6 +83,7 @@ file(GLOB_RECURSE trt_source plugin/*.cpp ) cuda_add_library(tinytrt SHARED ${trt_source}) +target_include_directories(tinytrt PUBLIC ./) target_compile_options(tinytrt PUBLIC -std=c++11 -Wall -Wno-deprecated -Wfloat-conversion) set_target_properties(tinytrt PROPERTIES POSITION_INDEPENDENT_CODE ON) set_target_properties(tinytrt PROPERTIES VERSION ${PROJECT_VERSION}) From dde788bd916c8934584118fb168d9d8e64e957bf Mon Sep 17 00:00:00 2001 From: Shiroki Satsuki Date: Wed, 6 Jul 2022 16:48:09 +0800 Subject: [PATCH 2/2] feat: link in CMakeLists.txt --- CMakeLists.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/CMakeLists.txt b/CMakeLists.txt index 3ac9cfa..843bc53 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -84,6 +84,7 @@ file(GLOB_RECURSE trt_source ) cuda_add_library(tinytrt SHARED ${trt_source}) target_include_directories(tinytrt PUBLIC ./) +target_link_libraries(tinytrt ${LIBNVINFER} ${LIBNVINFER_PLUGIN} ${LIBNVPARSERS} ${LIBNVONNXPARSER} ${ZLIB_LIBRARIES}) target_compile_options(tinytrt PUBLIC -std=c++11 -Wall -Wno-deprecated -Wfloat-conversion) set_target_properties(tinytrt PROPERTIES POSITION_INDEPENDENT_CODE ON) set_target_properties(tinytrt PROPERTIES VERSION ${PROJECT_VERSION})