Updated onnxrt_build.sh (adding --allow_running_as_root) and Dockerfile (pip install flatbuffers)

Author: junechul

Dockerfile

@@ -84,9 +84,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
         python3 -m pip install --upgrade pip; \
     fi && \
     if echo ${BASE_IMAGE} | grep -q "debian" || echo ${BASE_IMAGE} | grep -q "ubuntu:24.04"; then \
-        python3 -m pip install --break-system-packages numpy setuptools wheel pybind11 pytest; \
+        python3 -m pip install --break-system-packages numpy setuptools wheel pybind11 pytest flatbuffers; \
     else \
-        python3 -m pip install numpy setuptools wheel pybind11 pytest; \
+        python3 -m pip install numpy setuptools wheel pybind11 pytest flatbuffers; \
     fi && \
     rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
 

scripts/onnxrt_build.sh

@@ -36,7 +36,8 @@ SECONDS=0
 --cmake_extra_defines="CMAKE_TOOLCHAIN_FILE=$(pwd)/tool.cmake" \
 --path_to_protoc_exe $(pwd)/cmake/external/protoc-${protobuf_ver}-linux-aarch_64/bin/protoc \
 --use_tidl \
---build_wheel
+--build_wheel \
+--allow_running_as_root
 
 # chmod
 chmod -R a+w $WORK_DIR/workarea

scripts/onnxrt_package.sh

@@ -14,7 +14,7 @@ cd $WORK_DIR/workarea
 ## package into a tarball
 onnx_ver=$(get_yaml_value "onnxruntime" "onnx_ver")
 pkg_dist=${BASE_IMAGE//:/}
-DST_DIR="onnx-${onnx_ver}+${tidl_ver}-${pkg_dist}_aarch64"
+DST_DIR="onnx-${onnx_ver}-${pkg_dist}_aarch64"
 LIB_DIR="onnxruntime/build/Linux/Release"
 TARBALL="${DST_DIR}.tar.gz"