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 a90157011832c95fd993ecd296524be1c5e88a85 Author: Andrew Reusch <[email protected]> AuthorDate: Fri Jun 10 14:07:05 2022 -0700 include relevant extras --- docker/Dockerfile.ci_arm | 9 ++++++++- docker/Dockerfile.ci_gpu | 15 ++++++++++++++- docker/Dockerfile.ci_hexagon | 9 ++++++++- docker/Dockerfile.ci_i386 | 5 ++++- docker/Dockerfile.ci_qemu | 9 ++++++++- docker/Dockerfile.ci_wasm | 3 ++- 6 files changed, 44 insertions(+), 6 deletions(-) diff --git a/docker/Dockerfile.ci_arm b/docker/Dockerfile.ci_arm index d6cf6c9b64..a405f11467 100644 --- a/docker/Dockerfile.ci_arm +++ b/docker/Dockerfile.ci_arm @@ -56,7 +56,14 @@ RUN bash /install/ubuntu_install_cmake_source.sh COPY python/build/pyproject.toml /install/python/pyproject.toml COPY python/build/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 +RUN bash /install/ubuntu_install_python_package.sh \ + -E importer-caffe \ + -E importer-caffe2 \ + -E importer-tensorflow \ + -E importer-tflite \ + -E tvmc \ + ; + # TensorFlow C++ deps COPY install/ubuntu_install_tensorflow_aarch64.sh /install/ubuntu_install_tensorflow_aarch64.sh diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu index ca1defa1a0..65f96c28c9 100644 --- a/docker/Dockerfile.ci_gpu +++ b/docker/Dockerfile.ci_gpu @@ -51,7 +51,20 @@ RUN bash /install/ubuntu_install_opencl.sh COPY python/build/pyproject.toml /install/python/pyproject.toml COPY python/build/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 +RUN bash /install/ubuntu_install_python_package.sh \ + -E gpu \ + -E importer-coreml \ + -E importer-darknet \ + -E importer-keras \ + -E importer-oneflow \ + -E importer-onnx \ + -E importer-paddle \ + -E importer-pytorch \ + -E importer-tensorflow \ + -E importer-tflite \ + -E tvmc \ + -E xgboost \ + ; # COPY install/ubuntu_install_sphinx.sh /install/ubuntu_install_sphinx.sh # RUN bash /install/ubuntu_install_sphinx.sh diff --git a/docker/Dockerfile.ci_hexagon b/docker/Dockerfile.ci_hexagon index 6fc7a40dd2..780f043f02 100644 --- a/docker/Dockerfile.ci_hexagon +++ b/docker/Dockerfile.ci_hexagon @@ -44,7 +44,14 @@ ENV PATH $PATH:$CARGO_HOME/bin COPY python/build/pyproject.toml /install/python/pyproject.toml COPY python/build/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 +RUN bash /install/ubuntu_install_python_package.sh \ + -E importer-keras \ + -E importer-onnx \ + -E importer-tensorflow \ + -E importer-tflite \ + -E tvmc \ + ; + # ANTLR deps COPY install/ubuntu_install_java.sh /install/ubuntu_install_java.sh diff --git a/docker/Dockerfile.ci_i386 b/docker/Dockerfile.ci_i386 index 3eb1a71093..a08b5e7f9d 100644 --- a/docker/Dockerfile.ci_i386 +++ b/docker/Dockerfile.ci_i386 @@ -48,7 +48,10 @@ RUN bash /install/ubuntu_install_cmake_source.sh COPY python/build/pyproject.toml /install/python/pyproject.toml COPY python/build/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 +RUN bash /install/ubuntu_install_python_package.sh \ + -E tvmc \ + -E xgboost \ + ; # AutoTVM deps COPY install/ubuntu_install_redis.sh /install/ubuntu_install_redis.sh diff --git a/docker/Dockerfile.ci_qemu b/docker/Dockerfile.ci_qemu index 245aee9e8c..6341c55817 100644 --- a/docker/Dockerfile.ci_qemu +++ b/docker/Dockerfile.ci_qemu @@ -34,7 +34,14 @@ ENV PATH ${TVM_VENV}/bin:$PATH COPY python/build/pyproject.toml /install/python/pyproject.toml COPY python/build/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 +RUN bash /install/ubuntu_install_python_package.sh \ + -E ethosu \ + -E importer-keras \ + -E importer-onnx \ + -E importer-tensorflow \ + -E importer-tflite \ + -E tvmc \ + ; COPY install/ubuntu1804_install_llvm.sh /install/ubuntu1804_install_llvm.sh RUN bash /install/ubuntu1804_install_llvm.sh diff --git a/docker/Dockerfile.ci_wasm b/docker/Dockerfile.ci_wasm index b4394c6c7a..d1abf8965f 100644 --- a/docker/Dockerfile.ci_wasm +++ b/docker/Dockerfile.ci_wasm @@ -31,7 +31,8 @@ ENV PATH ${TVM_VENV}/bin:$PATH COPY python/build/pyproject.toml /install/python/pyproject.toml COPY python/build/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 +RUN bash /install/ubuntu_install_python_package.sh \ + ; # Rust env (build early; takes a while) COPY install/ubuntu_install_rust.sh /install/ubuntu_install_rust.sh
