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 b32315bb9a fix tensorflow deps
b32315bb9a is described below
commit b32315bb9a5411f9767f0002fa884af6066d2c4f
Author: Andrew Reusch <[email protected]>
AuthorDate: Mon Jun 13 18:04:05 2022 -0700
fix tensorflow deps
---
docker/Dockerfile.ci_arm | 9 ++++-----
docker/Dockerfile.ci_i386 | 4 ++++
...l_tensorflow_aarch64.sh => ubuntu_install_tensorflow_deps.sh} | 0
3 files changed, 8 insertions(+), 5 deletions(-)
diff --git a/docker/Dockerfile.ci_arm b/docker/Dockerfile.ci_arm
index a405f11467..e0e0b2e796 100644
--- a/docker/Dockerfile.ci_arm
+++ b/docker/Dockerfile.ci_arm
@@ -53,6 +53,10 @@ ENV PATH ${TVM_VENV}/bin:$PATH
COPY install/ubuntu_install_cmake_source.sh
/install/ubuntu_install_cmake_source.sh
RUN bash /install/ubuntu_install_cmake_source.sh
+# TensorFlow C++ deps
+COPY install/ubuntu_install_tensorflow_deps.sh
/install/ubuntu_install_tensorflow_deps.sh
+RUN bash /install/ubuntu_install_tensorflow_deps.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
@@ -64,11 +68,6 @@ RUN bash /install/ubuntu_install_python_package.sh \
-E tvmc \
;
-
-# TensorFlow C++ deps
-COPY install/ubuntu_install_tensorflow_aarch64.sh
/install/ubuntu_install_tensorflow_aarch64.sh
-RUN bash /install/ubuntu_install_tensorflow_aarch64.sh
-
# Caffe deps
COPY install/ubuntu_install_boost.sh /install/ubuntu_install_boost.sh
RUN bash /install/ubuntu_install_boost.sh
diff --git a/docker/Dockerfile.ci_i386 b/docker/Dockerfile.ci_i386
index a08b5e7f9d..7916ca7711 100644
--- a/docker/Dockerfile.ci_i386
+++ b/docker/Dockerfile.ci_i386
@@ -33,6 +33,10 @@ ENV RUSTUP_HOME /opt/rust
ENV CARGO_HOME /opt/rust
ENV PATH $PATH:$CARGO_HOME/bin
+# TensorFlow C++ deps
+COPY install/ubuntu_install_tensorflow_deps.sh
/install/ubuntu_install_tensorflow_deps.sh
+RUN bash /install/ubuntu_install_tensorflow_deps.sh
+
ENV TVM_VENV /venv/apache-tvm-py3.7
COPY python/bootstrap/lockfiles /install/python/bootstrap/lockfiles
COPY install/ubuntu1804_install_python.sh /install/ubuntu1804_install_python.sh
diff --git a/docker/install/ubuntu_install_tensorflow_aarch64.sh
b/docker/install/ubuntu_install_tensorflow_deps.sh
similarity index 100%
rename from docker/install/ubuntu_install_tensorflow_aarch64.sh
rename to docker/install/ubuntu_install_tensorflow_deps.sh