From 376593c2ce3d0ae980af1d796c90342929f2402e Mon Sep 17 00:00:00 2001 From: JuneChul Roh Date: Sat, 30 Nov 2024 15:59:08 -0600 Subject: [PATCH] Updated onnxrt_build.sh (adding --allow_running_as_root) and Dockerfile (pip install flatbuffers) --- .github/workflows/master-workflow-debian.yaml | 98 +++++++++---------- Dockerfile | 1 + scripts/onnxrt_build.sh | 3 +- 3 files changed, 52 insertions(+), 50 deletions(-) diff --git a/.github/workflows/master-workflow-debian.yaml b/.github/workflows/master-workflow-debian.yaml index 6c96cf9..eec8d7c 100644 --- 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 index 64110ff..9339b59 100644 --- 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 index 298ffcf..1d7a82c 100755 --- 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