Add imagination nna part in build_linux.sh. test=develop

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#*=}"