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 514c969d1500e534178256ed83e1f0ddbabb4f01 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 264d2a9272..3df656cc97 100644 --- a/docker/Dockerfile.ci_arm +++ b/docker/Dockerfile.ci_arm @@ -59,7 +59,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 471ecf9849..60d93aa0c8 100644 --- a/docker/Dockerfile.ci_gpu +++ b/docker/Dockerfile.ci_gpu @@ -53,7 +53,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 43e030792d..2a33fc119c 100644 --- a/docker/Dockerfile.ci_hexagon +++ b/docker/Dockerfile.ci_hexagon @@ -48,7 +48,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 ae06a01308..b499d977c2 100644 --- a/docker/Dockerfile.ci_i386 +++ b/docker/Dockerfile.ci_i386 @@ -54,7 +54,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 c54afa0671..78c17fa07f 100644 --- a/docker/Dockerfile.ci_qemu +++ b/docker/Dockerfile.ci_qemu @@ -36,7 +36,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 8de018b229..b4558bde61 100644 --- a/docker/Dockerfile.ci_wasm +++ b/docker/Dockerfile.ci_wasm @@ -33,7 +33,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
