From 300843659de262a7dda7866b5c44a11aef1d686a Mon Sep 17 00:00:00 2001 From: Jiansong Wang Date: Wed, 16 Sep 2020 17:54:16 +0800 Subject: [PATCH] Add imagination nna part in build_linux.sh. test=develop --- lite/tools/build_linux.sh | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/lite/tools/build_linux.sh b/lite/tools/build_linux.sh index 36306ad580..5c606b079e 100755 --- a/lite/tools/build_linux.sh +++ b/lite/tools/build_linux.sh @@ -27,6 +27,9 @@ WITH_OPENCL=OFF # options of compiling rockchip NPU lib. WITH_ROCKCHIP_NPU=OFF ROCKCHIP_NPU_SDK_ROOT="$(pwd)/rknpu_ddk" # Download RKNPU SDK from https://github.com/airockchip/rknpu_ddk.git +# options of compiling imagination NNA lib +WITH_IMAGINATION_NNA=ON +IMAGINATION_NNA_SDK_ROOT="$(pwd)/imagination_nna_sdk/" # options of compiling baidu XPU lib. WITH_BAIDU_XPU=OFF BAIDU_XPU_SDK_ROOT="" @@ -70,7 +73,10 @@ function init_cmake_mutable_options { -DRKNPU_DDK_ROOT=$ROCKCHIP_NPU_SDK_ROOT \ -DLITE_WITH_XPU=$WITH_BAIDU_XPU \ -DXPU_SDK_ROOT=$BAIDU_XPU_SDK_ROOT \ - -DLITE_WITH_TRAIN=$WITH_TRAIN" + -DLITE_WITH_TRAIN=$WITH_TRAIN \ + -DLITE_WITH_IMAGINATION_NNA=$WITH_IMAGINATION_NNA \ + -DIMAGINATION_NNA_SDK_ROOT=${IMAGINATION_NNA_SDK_ROOT}" + } ##################################################################################################### @@ -317,6 +323,15 @@ function main { ROCKCHIP_NPU_SDK_ROOT="${i#*=}" shift ;; + # compiling lib which can operate on imagination nna. + --with_imagination_nna=*) + WITH_IMAGINATION_NNA="${i#*=}" + shift + ;; + --imagination_nna_sdk_root=*) + IMAGINATION_NNA_SDK_ROOT="${i#*=}" + shift + ;; # compiling lib which can operate on baidu xpu. --with_baidu_xpu=*) WITH_BAIDU_XPU="${i#*=}" -- GitLab