mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

fix cmake

Browse Source
This commit is contained in:
Francesco Gatti
2019-06-28 17:44:32 +02:00
parent 4ebbb6af2b
commit 6656c3d0e8
3 changed files with 26 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
CMakeLists.txt
+3 -3
View File
@@ -1,15 +1,15 @@
cmake_minimum_required(VERSION 2.8) cmake_minimum_required(VERSION 2.8)
project (tkDNN) project (tkDNN)
set(BUILD_DEPS true CACHE BOOL "If true download deps") set(TEST_DATA true CACHE BOOL "If true download deps")
if( ${BUILD_DEPS} ) if( ${TEST_DATA} )
message("Launching pre-build dependency installer script...") message("Launching pre-build dependency installer script...")
execute_process (COMMAND bash -c "bash build_models.sh download" execute_process (COMMAND bash -c "bash build_models.sh download"
WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/tests) WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/tests)
set(BUILD_DEPS false CACHE BOOL "If true download deps" FORCE) set(TEST_DATA false CACHE BOOL "If true download deps" FORCE)
message("Finished dowloading test weights") message("Finished dowloading test weights")
endif() endif()
README.md
+1 -1
View File
@@ -20,7 +20,7 @@ Build with cmake
mkdir build mkdir build
cd build cd build
cmake .. cmake ..
# use -DBUILD_DEPS=False to skip dataset download # use -DTEST_DATA=False to skip dataset download
make make
``` ```
during the cmake configuration it will be dowloaded the weights needed for running during the cmake configuration it will be dowloaded the weights needed for running
tkDNNConfig.cmake
+22 -2
View File
@@ -1,3 +1,7 @@
message("-- Found tkDNN")
set(CMAKE_MODULE_PATH ${CMAKE_CURRENT_LIST_DIR})
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} --std=c++11 -fPIC")
find_package(CUDA REQUIRED) find_package(CUDA REQUIRED)
find_package(OpenCV REQUIRED) find_package(OpenCV REQUIRED)
find_library(NVINFER NAMES nvinfer) find_library(NVINFER NAMES nvinfer)
@@ -6,5 +10,21 @@ if(NVINFER STREQUAL "NVINFER-NOTFOUND")
link_directories(/usr/local/nvidia/tensorrt/targets/x86_64-linux-gnu/lib/ link_directories(/usr/local/nvidia/tensorrt/targets/x86_64-linux-gnu/lib/
/usr/local/cuda/targets/x86_64-linux/lib/) /usr/local/cuda/targets/x86_64-linux/lib/)
endif() endif()
set(tkDNN_INCLUDE_DIRS ${CUDA_INCLUDE_DIRS} ${OPENCV_INCLUDE_DIRS} ${NVINFER_INCLUDES})
set(tkDNN_LIBRARIES tkDNN kernels ${CUDA_LIBRARIES} ${CUDA_CUBLAS_LIBRARIES} -lcudnn -lnvinfer ${OpenCV_LIBS}) set(tkDNN_INCLUDE_DIRS
${CUDA_INCLUDE_DIRS}
${OPENCV_INCLUDE_DIRS}
${NVINFER_INCLUDES}
)
set(tkDNN_LIBRARIES
tkDNN
kernels
${CUDA_LIBRARIES}
${CUDA_CUBLAS_LIBRARIES}
-lcudnn
-lnvinfer
${OpenCV_LIBS}
)
set(tkDNN_FOUND true)