fix tensorrt cmakelist.txt (#264)

CoinCheung · Sep 8, 2022 · 7a6f123 · 7a6f123
1 parent 12c34e1
commit 7a6f123
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tensorrt/CMakeLists.txt b/tensorrt/CMakeLists.txt
@@ -15,6 +15,8 @@ link_directories(${PROJECT_SOURCE_DIR}/build)
 find_package(CUDA REQUIRED)
 find_package(OpenCV REQUIRED)
 
+cuda_add_library(kernels STATIC kernels.cu)
+
 add_executable(segment segment.cpp trt_dep.cpp)
 target_include_directories(
     segment PUBLIC ${CUDA_INCLUDE_DIRS} ${CUDNN_INCLUDE_DIRS} ${OpenCV_INCLUDE_DIRS})
@@ -23,4 +25,3 @@ target_link_libraries(
     ${CUDA_LIBRARIES}
     ${OpenCV_LIBRARIES})
 
-cuda_add_library(kernels STATIC kernels.cu)