diff --git a/lite/tools/build_linux.sh b/lite/tools/build_linux.sh index 36306ad5807006492733cebe81e5ebb18f84751f..5c606b079efcee4a21c92a9715622ed74a2f62d1 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#*=}"