This is an automated email from the ASF dual-hosted git repository.
mshr pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new ba0881ef24 [Docker][CI] Add NNEF dependency to CI images (#17433)
ba0881ef24 is described below
commit ba0881ef24d17a11d7a46e4d662cb4b1632a652c
Author: Ágoston Czobor <[email protected]>
AuthorDate: Fri Oct 4 07:55:26 2024 +0200
[Docker][CI] Add NNEF dependency to CI images (#17433)
[Docker][CI] Add NNEF dependency
---
docker/Dockerfile.ci_arm | 4 ++++
docker/Dockerfile.ci_cortexm | 4 ++++
docker/Dockerfile.ci_cpu | 4 ++++
docker/Dockerfile.ci_gpu | 3 +++
docker/Dockerfile.ci_hexagon | 4 ++++
docker/Dockerfile.ci_riscv | 4 ++++
docker/install/ubuntu_install_nnef.sh | 25 +++++++++++++++++++++++++
docker/python/ci-constraints.txt | 2 ++
8 files changed, 50 insertions(+)
diff --git a/docker/Dockerfile.ci_arm b/docker/Dockerfile.ci_arm
index 2be887079e..16ffecb315 100644
--- a/docker/Dockerfile.ci_arm
+++ b/docker/Dockerfile.ci_arm
@@ -75,6 +75,10 @@ RUN bash /install/ubuntu_install_tflite.sh
COPY install/ubuntu_install_onnx.sh /install/ubuntu_install_onnx.sh
RUN bash /install/ubuntu_install_onnx.sh
+# NNEF
+COPY install/ubuntu_install_nnef.sh /install/ubuntu_install_nnef.sh
+RUN bash /install/ubuntu_install_nnef.sh
+
# AutoTVM deps
COPY install/ubuntu_install_redis.sh /install/ubuntu_install_redis.sh
RUN bash /install/ubuntu_install_redis.sh
diff --git a/docker/Dockerfile.ci_cortexm b/docker/Dockerfile.ci_cortexm
index 8006b27e84..5535d29ed1 100644
--- a/docker/Dockerfile.ci_cortexm
+++ b/docker/Dockerfile.ci_cortexm
@@ -108,6 +108,10 @@ RUN bash /install/ubuntu_install_arduino.sh
COPY install/ubuntu_install_onnx.sh /install/ubuntu_install_onnx.sh
RUN bash /install/ubuntu_install_onnx.sh
+# NNEF
+COPY install/ubuntu_install_nnef.sh /install/ubuntu_install_nnef.sh
+RUN bash /install/ubuntu_install_nnef.sh
+
# Install CMSIS_NN
COPY install/ubuntu_install_cmsis.sh /install/ubuntu_install_cmsis.sh
RUN bash /install/ubuntu_install_cmsis.sh /opt/arm/ethosu/cmsis
diff --git a/docker/Dockerfile.ci_cpu b/docker/Dockerfile.ci_cpu
index 37c7c90857..9e53882e16 100644
--- a/docker/Dockerfile.ci_cpu
+++ b/docker/Dockerfile.ci_cpu
@@ -134,6 +134,10 @@ RUN bash /install/ubuntu_install_libxsmm.sh
COPY install/ubuntu_install_onnx.sh /install/ubuntu_install_onnx.sh
RUN bash /install/ubuntu_install_onnx.sh
+# NNEF
+COPY install/ubuntu_install_nnef.sh /install/ubuntu_install_nnef.sh
+RUN bash /install/ubuntu_install_nnef.sh
+
# AArch64 Architecture Envelope Model (AEM)
COPY install/ubuntu_install_aprofile_aem.sh /install
RUN bash /install/ubuntu_install_aprofile_aem.sh
diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu
index 1a5721c549..7f5a68911c 100644
--- a/docker/Dockerfile.ci_gpu
+++ b/docker/Dockerfile.ci_gpu
@@ -104,6 +104,9 @@ RUN bash /install/ubuntu_install_libtorch.sh
COPY install/ubuntu_install_tflite.sh /install/ubuntu_install_tflite.sh
RUN bash /install/ubuntu_install_tflite.sh
+COPY install/ubuntu_install_nnef.sh /install/ubuntu_install_nnef.sh
+RUN bash /install/ubuntu_install_nnef.sh
+
COPY install/ubuntu_install_dgl.sh /install/ubuntu_install_dgl.sh
RUN bash /install/ubuntu_install_dgl.sh
diff --git a/docker/Dockerfile.ci_hexagon b/docker/Dockerfile.ci_hexagon
index 11b3041f3c..489894d252 100644
--- a/docker/Dockerfile.ci_hexagon
+++ b/docker/Dockerfile.ci_hexagon
@@ -84,6 +84,10 @@ RUN bash /install/ubuntu_install_tflite.sh
COPY install/ubuntu_install_onnx.sh /install/ubuntu_install_onnx.sh
RUN bash /install/ubuntu_install_onnx.sh
+# NNEF
+COPY install/ubuntu_install_nnef.sh /install/ubuntu_install_nnef.sh
+RUN bash /install/ubuntu_install_nnef.sh
+
# xgboost (for tuning)
COPY install/ubuntu_install_redis.sh /install/ubuntu_install_redis.sh
RUN bash /install/ubuntu_install_redis.sh
diff --git a/docker/Dockerfile.ci_riscv b/docker/Dockerfile.ci_riscv
index d1b5a033b6..c26470985a 100644
--- a/docker/Dockerfile.ci_riscv
+++ b/docker/Dockerfile.ci_riscv
@@ -75,6 +75,10 @@ RUN bash /install/ubuntu_install_tflite.sh
COPY install/ubuntu_install_onnx.sh /install/ubuntu_install_onnx.sh
RUN bash /install/ubuntu_install_onnx.sh
+# NNEF
+COPY install/ubuntu_install_nnef.sh /install/ubuntu_install_nnef.sh
+RUN bash /install/ubuntu_install_nnef.sh
+
# sccache
COPY install/ubuntu_install_sccache.sh /install/ubuntu_install_sccache.sh
RUN bash /install/ubuntu_install_sccache.sh
diff --git a/docker/install/ubuntu_install_nnef.sh
b/docker/install/ubuntu_install_nnef.sh
new file mode 100644
index 0000000000..6cd4761787
--- /dev/null
+++ b/docker/install/ubuntu_install_nnef.sh
@@ -0,0 +1,25 @@
+#!/bin/bash
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+set -e
+set -u
+set -o pipefail
+
+pip3 install \
+ nnef_tools==1.0.6 \
+ nnef==1.0.7
diff --git a/docker/python/ci-constraints.txt b/docker/python/ci-constraints.txt
index 003c131704..feba27cd03 100644
--- a/docker/python/ci-constraints.txt
+++ b/docker/python/ci-constraints.txt
@@ -37,3 +37,5 @@ tflite = "==2.4.0"
torch = "==1.11.0"
torchvision = "==0.12.0+cpu"
#xgboost = "==1.4.2"
+nnef = "==1.0.7"
+nnef_tools = "==1.0.6"
