From 68a19a73681301c6712e10d55bc64324716dbd24 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期三, 09 十月 2019 15:38:47 +0800
Subject: [PATCH] split scale
---
CMakeLists.txt | 8 +++++++-
1 files changed, 7 insertions(+), 1 deletions(-)
diff --git a/CMakeLists.txt b/CMakeLists.txt
index 5ef89c8..6e1e6b5 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -22,7 +22,12 @@
${CMAKE_SOURCE_DIR}/csrc/thirdparty/whereami
${CMAKE_SOURCE_DIR}/csrc/thirdparty/ffmpeg/include
${CMAKE_SOURCE_DIR}/csrc/thirdparty/gb28181/include
+ ${CMAKE_SOURCE_DIR}/csrc/gpu-conv
)
+
+set(CUDA_TOOLKIT_ROOT_DIR /usr/local/cuda)
+include_directories(/usr/local/cuda/include)
+find_package(CUDA QUIET REQUIRED)
link_directories(/usr/local/cuda/lib64 ${CMAKE_SOURCE_DIR}/csrc/thirdparty/gb28181/lib)
@@ -47,7 +52,8 @@
file(GLOB_RECURSE FFMPEG_LIST ${CMAKE_SOURCE_DIR}/csrc/*.cpp)
list(APPEND FFMPEG_LIST ${CMAKE_SOURCE_DIR}/csrc/thirdparty/whereami/whereami.c)
+file(GLOB_RECURSE CUDA_LIST ${CMAKE_SOURCE_DIR}/csrc/gpu-conv/*.cu)
-add_library(${BIN} SHARED ${FFMPEG_LIST})
+cuda_add_library(${BIN} SHARED ${FFMPEG_LIST} ${CUDA_LIST})
target_link_libraries(${BIN} ${LINK_LIB} numa nppig nppicc nppc -lz pthread dl rtspclient StreamParse)
--
Gitblit v1.8.0