onnx-mlir/docker/Dockerfile

19 lines
513 B
Docker

# Grab pre-built LLVM from github packages
FROM onnxmlirczar/onnx-mlir-llvmimage:initial
WORKDIR /build/bin
WORKDIR /build
ENV HOME=/build
ENV PYENV_ROOT=$HOME/.pyenv
ENV PATH=$PYENV_ROOT/shims:$PYENV_ROOT/bin:$PATH
RUN pyenv global 3.7.0
RUN pyenv rehash
ENV PATH=$PATH:/build/bin
RUN git clone --recursive https://github.com/onnx/onnx-mlir.git
COPY compile-onnx-mlir.sh bin/compile-onnx-mlir.sh
RUN compile-onnx-mlir.sh
WORKDIR /build/onnx-mlir/build
ENV LIT_OPT=v
RUN cmake --build . --target check-onnx-lit