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

auto download

Browse Source
This commit is contained in:
Francesco Gatti
2017-08-09 14:13:17 +00:00
parent 3215d5aab0
commit 1c6888f312
2 changed files with 25 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files
CMakeLists.txt
+13 -2
View File
@@ -1,7 +1,18 @@
cmake_minimum_required(VERSION 2.8)
project (tkDNN)
set(BUILD_DEPS true CACHE BOOL "If true download deps")
if( ${BUILD_DEPS} )
message("Launching pre-build dependency installer script...")
execute_process (COMMAND bash -c "bash build_models.sh download"
WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/tests)
set(BUILD_DEPS false CACHE BOOL "If true download deps" FORCE)
message("Finished dowloading test weights")
endif()
find_package(CUDA QUIET REQUIRED)
cuda_include_directories(${CMAKE_CURRENT_SOURCE_DIR}/include ${CUDA_INCLUDE_DIRS})
@@ -31,4 +42,4 @@ add_executable(test_yolo tests/yolo/yolo.cpp)
target_link_libraries(test_yolo tkDNN)
add_executable(test_yolo_tiny tests/yolo-tiny/yolo-tiny.cpp)
target_link_libraries(test_yolo_tiny tkDNN)
target_link_libraries(test_yolo_tiny tkDNN)