Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
143 changes: 0 additions & 143 deletions .github/workflows/ax.yml

This file was deleted.

28 changes: 1 addition & 27 deletions .github/workflows/ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -223,30 +223,4 @@ jobs:
tags: ghcr.io/${{ steps.lowercaseRepo.outputs.lowercase }}:${{ env.SHORT_SHA }}
sources: |
ghcr.io/${{ steps.lowercaseRepo.outputs.lowercase }}:${{ env.SHORT_SHA }}-amd64
ghcr.io/${{ steps.lowercaseRepo.outputs.lowercase }}:${{ env.SHORT_SHA }}-rpi
axera_build:
runs-on: ubuntu-22.04
name: AXERA Build
needs:
- amd64_build
- arm64_build
steps:
- name: Check out code
uses: actions/checkout@v5
with:
persist-credentials: false
- name: Set up QEMU and Buildx
id: setup
uses: ./.github/actions/setup
with:
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
- name: Build and push Axera build
uses: docker/bake-action@v6
with:
source: .
push: true
targets: axcl
files: docker/axcl/axcl.hcl
set: |
axcl.tags=${{ steps.setup.outputs.image-name }}-axcl
*.cache-from=type=gha
ghcr.io/${{ steps.lowercaseRepo.outputs.lowercase }}:${{ env.SHORT_SHA }}-rpi
25 changes: 0 additions & 25 deletions docker/axcl/Dockerfile

This file was deleted.

13 changes: 0 additions & 13 deletions docker/axcl/axcl.hcl

This file was deleted.

15 changes: 0 additions & 15 deletions docker/axcl/axcl.mk

This file was deleted.

7 changes: 0 additions & 7 deletions docker/axcl/rk-axcl.hcl

This file was deleted.

7 changes: 0 additions & 7 deletions docker/axcl/rpi-axcl.hcl

This file was deleted.

13 changes: 0 additions & 13 deletions docker/axcl/x86-axcl.hcl

This file was deleted.

10 changes: 10 additions & 0 deletions docker/main/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -266,6 +266,16 @@ RUN wget -q https://bootstrap.pypa.io/get-pip.py -O get-pip.py \
RUN --mount=type=bind,from=wheels,source=/wheels,target=/deps/wheels \
pip3 install -U /deps/wheels/*.whl

####
#
# AXEngine Support
#
####
RUN pip3 install https://github.com/ivanshi1108/pyaxengine/releases/download/0.1.3-frigate/axengine-0.1.3-py3-none-any.whl

ENV PATH="${PATH}:/usr/bin/axcl"
ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/usr/lib/axcl"

# Install MemryX runtime (requires libgomp (OpenMP) in the final docker image)
RUN --mount=type=bind,source=docker/main/install_memryx.sh,target=/deps/install_memryx.sh \
bash -c "bash /deps/install_memryx.sh"
Expand Down
9 changes: 5 additions & 4 deletions frigate/detectors/detection_runners.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,10 +10,6 @@
import numpy as np
import onnxruntime as ort

from frigate.util.axengine_converter import (
auto_convert_model as auto_load_axengine_model,
)
from frigate.util.axengine_converter import is_axengine_compatible
from frigate.util.model import get_ort_providers
from frigate.util.rknn_converter import auto_convert_model, is_rknn_compatible

Expand Down Expand Up @@ -675,6 +671,11 @@ def get_optimized_runner(
"""Get an optimized runner for the hardware."""
device = device or "AUTO"

from frigate.util.axengine_converter import (
auto_convert_model as auto_load_axengine_model,
)
from frigate.util.axengine_converter import is_axengine_compatible

if is_axengine_compatible(model_path, device, model_type):
axmodel_path = auto_load_axengine_model(model_path, model_type)

Expand Down