This is an automated email from the ASF dual-hosted git repository. areusch pushed a commit to branch areusch/freeze-dependencies in repository https://gitbox.apache.org/repos/asf/tvm.git
commit 53dfdea8e89f69def421ead659621e7ab5897692 Author: Andrew Reusch <[email protected]> AuthorDate: Wed May 11 07:59:49 2022 -0700 changes to dockerfile and pyproject --- docker/Dockerfile.base_arm64 | 7 +++++++ docker/Dockerfile.base_x86_64 | 14 ++++++++++++-- docker/Dockerfile.ci_cpu | 35 ++++++++++++++--------------------- docker/Dockerfile.ci_gpu | 36 ++++++++++++++++++------------------ pyproject.toml | 15 +++++++++++++++ 5 files changed, 66 insertions(+), 41 deletions(-) diff --git a/docker/Dockerfile.base_arm64 b/docker/Dockerfile.base_arm64 index 31a43c2edc..4cc9cb7ec2 100644 --- a/docker/Dockerfile.base_arm64 +++ b/docker/Dockerfile.base_arm64 @@ -26,6 +26,13 @@ RUN apt-get install -y ca-certificates gnupg2 COPY install/ubuntu_install_core.sh /install/ubuntu_install_core.sh RUN bash /install/ubuntu_install_core.sh +# Rust env (build early; takes a while). Needed for some python packages (e.g. cryptography). +COPY install/ubuntu_install_rust.sh /install/ubuntu_install_rust.sh +RUN bash /install/ubuntu_install_rust.sh +ENV RUSTUP_HOME /opt/rust +ENV CARGO_HOME /opt/rust +ENV PATH $PATH:$CARGO_HOME/bin + COPY install/ubuntu1804_install_python.sh /install/ubuntu1804_install_python.sh RUN bash /install/ubuntu1804_install_python.sh diff --git a/docker/Dockerfile.base_x86_64 b/docker/Dockerfile.base_x86_64 index eb2c84454c..3bed3676a5 100644 --- a/docker/Dockerfile.base_x86_64 +++ b/docker/Dockerfile.base_x86_64 @@ -18,9 +18,19 @@ # CI docker CPU env FROM ubuntu:18.04 +COPY install/ubuntu_install_core.sh /install/ubuntu_install_core.sh +RUN bash /install/ubuntu_install_core.sh + +# Rust env (build early; takes a while). Needed for some python packages (e.g. cryptography). +COPY install/ubuntu_install_rust.sh /install/ubuntu_install_rust.sh +RUN bash /install/ubuntu_install_rust.sh +ENV RUSTUP_HOME /opt/rust +ENV CARGO_HOME /opt/rust +ENV PATH $PATH:$CARGO_HOME/bin + COPY python/bootstrap-requirements-x86_64.txt /install/python/bootstrap-requirements.txt COPY install/ubuntu1804_install_python.sh /install/ubuntu1804_install_python.sh RUN bash /install/ubuntu1804_install_python.sh -COPY install/ubuntu_install_core.sh /install/ubuntu_install_core.sh -RUN bash /install/ubuntu_install_core.sh +# Globally disable pip cache +RUN pip config set global.no-cache-dir false diff --git a/docker/Dockerfile.ci_cpu b/docker/Dockerfile.ci_cpu index 40a6eedab7..8ce41386a8 100644 --- a/docker/Dockerfile.ci_cpu +++ b/docker/Dockerfile.ci_cpu @@ -16,10 +16,10 @@ # under the License. # CI docker CPU env -FROM tvm_ci.base_x86_64:latest +FROM tvm.base_x86_64:latest -COPY install/build/base_x86_64/pyproject.toml /install/python/pyproject.toml -COPY install/build/base_x86_64/poetry.lock /install/python/poetry.lock +COPY build/base_x86_64/pyproject.toml /install/python/pyproject.toml +COPY build/base_x86_64/poetry.lock /install/python/poetry.lock COPY install/ubuntu_install_python_package.sh /install/ubuntu_install_python_package.sh RUN bash /install/ubuntu_install_python_package.sh @@ -32,16 +32,9 @@ RUN bash /install/ubuntu_install_dnnl.sh COPY install/ubuntu_install_papi.sh /install/ubuntu_install_papi.sh RUN bash /install/ubuntu_install_papi.sh "" -# Install MxNet for access to Gluon Model Zoo. -COPY install/ubuntu_install_mxnet.sh /install/ubuntu_install_mxnet.sh -RUN bash /install/ubuntu_install_mxnet.sh - -# Rust env (build early; takes a while) -COPY install/ubuntu_install_rust.sh /install/ubuntu_install_rust.sh -RUN bash /install/ubuntu_install_rust.sh -ENV RUSTUP_HOME /opt/rust -ENV CARGO_HOME /opt/rust -ENV PATH $PATH:$CARGO_HOME/bin +# # Install MxNet for access to Gluon Model Zoo. +# COPY install/ubuntu_install_mxnet.sh /install/ubuntu_install_mxnet.sh +# RUN bash /install/ubuntu_install_mxnet.sh # wasmtime COPY install/ubuntu_install_wasmtime.sh /install/ubuntu_install_wasmtime.sh @@ -77,12 +70,12 @@ COPY install/ubuntu_install_verilator.sh /install/ubuntu_install_verilator.sh RUN bash /install/ubuntu_install_verilator.sh # TensorFlow deps -COPY install/ubuntu_install_tensorflow.sh /install/ubuntu_install_tensorflow.sh -RUN bash /install/ubuntu_install_tensorflow.sh +# COPY install/ubuntu_install_tensorflow.sh /install/ubuntu_install_tensorflow.sh +# RUN bash /install/ubuntu_install_tensorflow.sh # TFLite deps -COPY install/ubuntu_install_tflite.sh /install/ubuntu_install_tflite.sh -RUN bash /install/ubuntu_install_tflite.sh +# COPY install/ubuntu_install_tflite.sh /install/ubuntu_install_tflite.sh +# RUN bash /install/ubuntu_install_tflite.sh # Compute Library COPY install/ubuntu_download_arm_compute_lib_binaries.sh /install/ubuntu_download_arm_compute_lib_binaries.sh @@ -120,15 +113,15 @@ COPY install/ubuntu_install_ethosu_driver_stack.sh /install/ubuntu_install_ethos RUN bash /install/ubuntu_install_ethosu_driver_stack.sh # Install Vela compiler -COPY install/ubuntu_install_vela.sh /install/ubuntu_install_vela.sh -RUN bash /install/ubuntu_install_vela.sh +# COPY install/ubuntu_install_vela.sh /install/ubuntu_install_vela.sh +# RUN bash /install/ubuntu_install_vela.sh # Update PATH ENV PATH /opt/arm/gcc-arm-none-eabi/bin:/opt/arm/FVP_Corstone_SSE-300/models/Linux64_GCC-6.4:$PATH # PaddlePaddle deps -COPY install/ubuntu_install_paddle.sh /install/ubuntu_install_paddle.sh -RUN bash /install/ubuntu_install_paddle.sh +# COPY install/ubuntu_install_paddle.sh /install/ubuntu_install_paddle.sh +# RUN bash /install/ubuntu_install_paddle.sh # sccache COPY install/ubuntu_install_sccache.sh /install/ubuntu_install_sccache.sh diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu index 5d0a642d3f..5a7db5315f 100644 --- a/docker/Dockerfile.ci_gpu +++ b/docker/Dockerfile.ci_gpu @@ -51,8 +51,8 @@ RUN bash /install/ubuntu_install_opencl.sh COPY install/ubuntu_install_python_package.sh /install/ubuntu_install_python_package.sh RUN bash /install/ubuntu_install_python_package.sh -COPY install/ubuntu_install_sphinx.sh /install/ubuntu_install_sphinx.sh -RUN bash /install/ubuntu_install_sphinx.sh +# COPY install/ubuntu_install_sphinx.sh /install/ubuntu_install_sphinx.sh +# RUN bash /install/ubuntu_install_sphinx.sh # Enable doxygen for c++ doc build RUN apt-get update && apt-get install -y doxygen libprotobuf-dev protobuf-compiler @@ -67,26 +67,26 @@ COPY install/ubuntu_install_rocm.sh /install/ubuntu_install_rocm.sh RUN bash /install/ubuntu_install_rocm.sh # DL Frameworks -COPY install/ubuntu_install_mxnet.sh /install/ubuntu_install_mxnet.sh -RUN bash /install/ubuntu_install_mxnet.sh +# COPY install/ubuntu_install_mxnet.sh /install/ubuntu_install_mxnet.sh +# RUN bash /install/ubuntu_install_mxnet.sh COPY install/ubuntu_install_gluoncv.sh /install/ubuntu_install_gluoncv.sh RUN bash /install/ubuntu_install_gluoncv.sh -COPY install/ubuntu_install_coreml.sh /install/ubuntu_install_coreml.sh -RUN bash /install/ubuntu_install_coreml.sh +# COPY install/ubuntu_install_coreml.sh /install/ubuntu_install_coreml.sh +# RUN bash /install/ubuntu_install_coreml.sh -COPY install/ubuntu_install_tensorflow.sh /install/ubuntu_install_tensorflow.sh -RUN bash /install/ubuntu_install_tensorflow.sh +# COPY install/ubuntu_install_tensorflow.sh /install/ubuntu_install_tensorflow.sh +# RUN bash /install/ubuntu_install_tensorflow.sh -COPY install/ubuntu_install_darknet.sh /install/ubuntu_install_darknet.sh -RUN bash /install/ubuntu_install_darknet.sh +# COPY install/ubuntu_install_darknet.sh /install/ubuntu_install_darknet.sh +# RUN bash /install/ubuntu_install_darknet.sh -COPY install/ubuntu_install_onnx.sh /install/ubuntu_install_onnx.sh -RUN bash /install/ubuntu_install_onnx.sh +# COPY install/ubuntu_install_onnx.sh /install/ubuntu_install_onnx.sh +# RUN bash /install/ubuntu_install_onnx.sh -COPY install/ubuntu_install_tflite.sh /install/ubuntu_install_tflite.sh -RUN bash /install/ubuntu_install_tflite.sh +# COPY install/ubuntu_install_tflite.sh /install/ubuntu_install_tflite.sh +# RUN bash /install/ubuntu_install_tflite.sh COPY install/ubuntu_install_dgl.sh /install/ubuntu_install_dgl.sh RUN bash /install/ubuntu_install_dgl.sh @@ -95,12 +95,12 @@ ENV NVIDIA_DRIVER_CAPABILITIES compute,graphics,utility COPY install/ubuntu_install_vulkan.sh /install/ubuntu_install_vulkan.sh RUN bash /install/ubuntu_install_vulkan.sh -COPY install/ubuntu_install_paddle.sh /install/ubuntu_install_paddle.sh -RUN bash /install/ubuntu_install_paddle.sh +# COPY install/ubuntu_install_paddle.sh /install/ubuntu_install_paddle.sh +# RUN bash /install/ubuntu_install_paddle.sh # OneFlow deps -COPY install/ubuntu_install_oneflow.sh /install/ubuntu_install_oneflow.sh -RUN bash /install/ubuntu_install_oneflow.sh +# COPY install/ubuntu_install_oneflow.sh /install/ubuntu_install_oneflow.sh +# RUN bash /install/ubuntu_install_oneflow.sh # Rust env (build early; takes a while) COPY install/ubuntu_install_rust.sh /install/ubuntu_install_rust.sh diff --git a/pyproject.toml b/pyproject.toml index 843d192d7a..65444820e0 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -52,3 +52,18 @@ name = "apache-tvm" authors = [] version = "0.8.0" description = "Open source Deep Learning compliation toolkit" + +[[tool.poetry.source]] +name = "oneflow" +url = "https://release.oneflow.info"; +secondary = true + +#[[tool.poetry.source]] +#name = "onnx" +#url = "https://download.pytorch.org/whl/cpu"; +#secondary = true + +[[tool.poetry.source]] +name = "tensorflow-aarch64" +url = "https://snapshots.linaro.org/ldcg/python-cache"; +secondary = true
