You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by ar...@apache.org on 2022/06/13 23:43:58 UTC
[tvm] 26/27: include relevant extras
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 a3c5df91a40350a3ba4d777a4b1a40d342d8773d
Author: Andrew Reusch <ar...@gmail.com>
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