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/07/12 21:48:11 UTC
[tvm] 26/47: 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 0dbbb0e21635bacf63592faf3f21bba4db95e733
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 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 ff8c2dcdd7..6076853b12 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