Updated onnxrt_build.sh (adding --allow_running_as_root) and Dockerfi…

…le (pip install flatbuffers)
TexasInstruments-Sandbox · Nov 30, 2024 · 376593c · 376593c
1 parent 9b39876
commit 376593c
Show file tree

Hide file tree

Showing 3 changed files with 52 additions and 50 deletions.
diff --git a/.github/workflows/master-workflow-debian.yaml b/.github/workflows/master-workflow-debian.yaml
@@ -14,56 +14,56 @@ jobs:
       base_image: "debian:12.5"
       sdk_ver: 10.1.0
 
-  ## Library build: Debian 12.5
-  onnxrt_build_debian12-5:
-    needs: docker_build_debian12-5
-    uses: ./.github/workflows/lib-onnxrt-build.yaml
-    secrets:
-      env_PAT: ${{ secrets.GHCR_PAT }}
-    with:
-      arch: arm64
-      base_image: "debian:12.5"
-      sdk_ver: 10.1.0
+  # ## Library build: Debian 12.5
+  # onnxrt_build_debian12-5:
+  #   needs: docker_build_debian12-5
+  #   uses: ./.github/workflows/lib-onnxrt-build.yaml
+  #   secrets:
+  #     env_PAT: ${{ secrets.GHCR_PAT }}
+  #   with:
+  #     arch: arm64
+  #     base_image: "debian:12.5"
+  #     sdk_ver: 10.1.0
 
-  tflite_build_debian12-5:
-    needs: docker_build_debian12-5
-    uses: ./.github/workflows/lib-tflite-build.yaml
-    secrets:
-      env_PAT: ${{ secrets.GHCR_PAT }}
-    with:
-      arch: arm64
-      base_image: "debian:12.5"
-      sdk_ver: 10.1.0
+  # tflite_build_debian12-5:
+  #   needs: docker_build_debian12-5
+  #   uses: ./.github/workflows/lib-tflite-build.yaml
+  #   secrets:
+  #     env_PAT: ${{ secrets.GHCR_PAT }}
+  #   with:
+  #     arch: arm64
+  #     base_image: "debian:12.5"
+  #     sdk_ver: 10.1.0
 
-  dlr_build_debian12-5:
-    needs: docker_build_debian12-5
-    uses: ./.github/workflows/lib-dlr-build.yaml
-    secrets:
-      env_PAT: ${{ secrets.GHCR_PAT }}
-    with:
-      arch: arm64
-      base_image: "debian:12.5"
-      sdk_ver: 10.1.0
+  # dlr_build_debian12-5:
+  #   needs: docker_build_debian12-5
+  #   uses: ./.github/workflows/lib-dlr-build.yaml
+  #   secrets:
+  #     env_PAT: ${{ secrets.GHCR_PAT }}
+  #   with:
+  #     arch: arm64
+  #     base_image: "debian:12.5"
+  #     sdk_ver: 10.1.0
 
-  tidl_build_debian12-5:
-    needs: docker_build_debian12-5
-    uses: ./.github/workflows/lib-tidl-build.yaml
-    secrets:
-      env_PAT: ${{ secrets.GHCR_PAT }}
-    with:
-      arch: arm64
-      base_image: "debian:12.5"
-      sdk_ver: 10.1.0
+  # tidl_build_debian12-5:
+  #   needs: docker_build_debian12-5
+  #   uses: ./.github/workflows/lib-tidl-build.yaml
+  #   secrets:
+  #     env_PAT: ${{ secrets.GHCR_PAT }}
+  #   with:
+  #     arch: arm64
+  #     base_image: "debian:12.5"
+  #     sdk_ver: 10.1.0
 
-  ## Release
-  release_debian12-5:
-    uses: ./.github/workflows/release-debian.yaml
-    needs:
-      - onnxrt_build_debian12-5
-      - tflite_build_debian12-5
-      - dlr_build_debian12-5
-      - tidl_build_debian12-5
-    secrets:
-      gh_TOKEN: ${{ secrets.GHCR_PAT }}
-    with:
-      sdk_ver: 10.1.0
+  # ## Release
+  # release_debian12-5:
+  #   uses: ./.github/workflows/release-debian.yaml
+  #   needs:
+  #     - onnxrt_build_debian12-5
+  #     - tflite_build_debian12-5
+  #     - dlr_build_debian12-5
+  #     - tidl_build_debian12-5
+  #   secrets:
+  #     gh_TOKEN: ${{ secrets.GHCR_PAT }}
+  #   with:
+  #     sdk_ver: 10.1.0
diff --git a/Dockerfile b/Dockerfile
@@ -88,6 +88,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     else \
         python3 -m pip install numpy setuptools wheel pybind11 pytest; \
     fi && \
+    python3 -m pip install flatbuffers && \
     rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
 
 # cmake >= 3.24

diff --git a/scripts/onnxrt_build.sh b/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