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
The following commit(s) were added to refs/heads/areusch/freeze-dependencies by
this push:
new 9b05f70862 include relevant extras
9b05f70862 is described below
commit 9b05f70862598edbe409965612ef82aa26c0c97a
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
