enhance armnn conversion

This commit is contained in:
mertalev
2024-07-05 19:38:47 -04:00
parent 5748f50c1f
commit 956480ab2c
21 changed files with 656 additions and 189 deletions

View File

@@ -0,0 +1,28 @@
FROM mambaorg/micromamba:bookworm-slim@sha256:333f7598ff2c2400fb10bfe057709c68b7daab5d847143af85abcf224a07271a as builder
ENV TRANSFORMERS_CACHE=/cache \
PYTHONDONTWRITEBYTECODE=1 \
PYTHONUNBUFFERED=1 \
PATH="/opt/venv/bin:$PATH"
WORKDIR /export/ann
USER root
RUN apt-get update && apt-get install -y --no-install-recommends \
build-essential \
curl \
git
USER $MAMBA_USER
COPY --chown=$MAMBA_USER:$MAMBA_USER env.yaml ./
RUN micromamba install -y -f env.yaml
COPY --chown=$MAMBA_USER:$MAMBA_USER *.sh *.cpp ./
ENV ARMNN_PATH=/export/ann/armnn
RUN ./download-armnn.sh && \
./build-converter.sh && \
./build.sh
COPY --chown=$MAMBA_USER:$MAMBA_USER run.py ./
ENTRYPOINT ["/usr/local/bin/_entrypoint.sh"]
CMD ["python", "run.py"]