From ca0feb41d92b9f017e436496c074ae97b3e1a20c Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=A2=85=ED=98=84/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Senior=20Engineer/=EC=82=BC=EC=84=B1?= =?utf8?q?=EC=A0=84=EC=9E=90?= Date: Wed, 2 May 2018 19:59:59 +0900 Subject: [PATCH] Add 'CaffeSource' an an external module (#190) This commit adds 'CaffeSource' external module which downloads Caffe source from the URL specified by CAFFE_URL environment variable. Signed-off-by: Jonghyun Park --- CMakeLists.txt | 1 + cmake/packages/CaffeSourceConfig.cmake | 18 ++++++++++++++++++ 2 files changed, 19 insertions(+) create mode 100644 cmake/packages/CaffeSourceConfig.cmake diff --git a/CMakeLists.txt b/CMakeLists.txt index 98d5fed..fbc463d 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -19,6 +19,7 @@ option(DOWNLOAD_GEMMLOWP "Download GEMM low precesion library source" OFF) option(DOWNLOAD_NEON2SSE "Download NEON2SSE library source" OFF) option(DOWNLOAD_FLATBUFFERS "Download FlatBuffers source" OFF) option(DOWNLOAD_TENSORFLOW "Download TensorFlow source" OFF) +option(DOWNLOAD_CAFFE "Download Caffe source" OFF) option(BUILD_GTEST "Download and build Google Test" ON) diff --git a/cmake/packages/CaffeSourceConfig.cmake b/cmake/packages/CaffeSourceConfig.cmake new file mode 100644 index 0000000..177e9ab --- /dev/null +++ b/cmake/packages/CaffeSourceConfig.cmake @@ -0,0 +1,18 @@ +function(_CaffeSource_import) + if(NOT DOWNLOAD_CAFFE) + set(CaffeSource_FOUND FALSE PARENT_SCOPE) + return() + endif(NOT DOWNLOAD_CAFFE) + + nncc_find_package(ExternalSourceTools REQUIRED) + nncc_find_package(OptionTools REQUIRED) + + envoption(CAFFE_URL https://github.com/BVLC/caffe/archive/1.0.tar.gz) + + ExternalSource_Download(CAFFE ${CAFFE_URL}) + + set(CaffeSource_DIR ${CAFFE_SOURCE_DIR} PARENT_SCOPE) + set(CaffeSource_FOUND ${DOWNLOAD_CAFFE} PARENT_SCOPE) +endfunction(_CaffeSource_import) + +_CaffeSource_import() -- 2.7.4