This is an automated email from the ASF dual-hosted git repository.
masahi 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 8b59f99 [CI][VitisAI] Update CI Vitis AI PyXIR version to v0.3.1
(#8814)
8b59f99 is described below
commit 8b59f9995f4b80decf464ba40d6b2c8f75d87b36
Author: Jorn Tuyls <[email protected]>
AuthorDate: Sat Sep 11 07:28:57 2021 +0200
[CI][VitisAI] Update CI Vitis AI PyXIR version to v0.3.1 (#8814)
* Update CI Vitis AI PyXIR version to v0.3.1
* Add Vitis AI requirements to gen_requirements.py
---
docker/install/ubuntu_install_vitis_ai_packages_ci.sh | 2 +-
python/gen_requirements.py | 13 +++++++++++++
.../python/contrib/test_vitis_ai/test_vitis_ai_codegen.py | 14 --------------
3 files changed, 14 insertions(+), 15 deletions(-)
diff --git a/docker/install/ubuntu_install_vitis_ai_packages_ci.sh
b/docker/install/ubuntu_install_vitis_ai_packages_ci.sh
index 774d85d..25c2140 100644
--- a/docker/install/ubuntu_install_vitis_ai_packages_ci.sh
+++ b/docker/install/ubuntu_install_vitis_ai_packages_ci.sh
@@ -25,5 +25,5 @@ mkdir "$PYXIR_HOME"
pip3 install progressbar h5py==2.10.0
-git clone --recursive --branch v0.1.6 --depth 1
https://github.com/Xilinx/pyxir.git "${PYXIR_HOME}"
+git clone --recursive --branch v0.3.1 --depth 1
https://github.com/Xilinx/pyxir.git "${PYXIR_HOME}"
cd "${PYXIR_HOME}" && python3 setup.py install
diff --git a/python/gen_requirements.py b/python/gen_requirements.py
index 781db2c..a9a8607 100755
--- a/python/gen_requirements.py
+++ b/python/gen_requirements.py
@@ -150,6 +150,17 @@ REQUIREMENTS_BY_PIECE: RequirementsByPieceType = [
],
),
),
+ # Vitis AI requirements
+ (
+ "vitis-ai",
+ (
+ "Requirements for the Vitis AI codegen",
+ [
+ "h5py",
+ "progressbar",
+ ],
+ ),
+ ),
# XGBoost, useful for autotuning on some targets.
(
"xgboost",
@@ -217,6 +228,7 @@ CONSTRAINTS = [
), # Work around
https://github.com/readthedocs/sphinx_rtd_theme/issues/1115
("ethos-u-vela", "==2.1.1"),
("future", None),
+ ("h5py", "==2.10.0"),
("image", None),
("matplotlib", None),
("numpy", None),
@@ -224,6 +236,7 @@ CONSTRAINTS = [
("onnxruntime", None),
("opencv-python", None),
("pillow", None),
+ ("progressbar", None),
("psutil", None),
("pylint", None),
("scipy", None),
diff --git a/tests/python/contrib/test_vitis_ai/test_vitis_ai_codegen.py
b/tests/python/contrib/test_vitis_ai/test_vitis_ai_codegen.py
index 2e16792..fe063c2 100644
--- a/tests/python/contrib/test_vitis_ai/test_vitis_ai_codegen.py
+++ b/tests/python/contrib/test_vitis_ai/test_vitis_ai_codegen.py
@@ -97,19 +97,6 @@ def test_bias_add():
verify_codegen(mod, params=params, dpu_target="DPUCZDX8G-zcu104")
-def test_relu():
- """Test relu operator for Vitis-AI DPUCADX8G and DPUCZDX8G-zcu104
targets"""
-
- shape = (10, 10)
- x = relay.var("x", shape=shape)
- y = relay.nn.relu(x)
- func = relay.Function([x], y)
- mod = tvm.IRModule()
- mod["main"] = func
- verify_codegen(mod, dpu_target="DPUCADX8G")
- verify_codegen(mod, dpu_target="DPUCZDX8G-zcu104")
-
-
def test_batchnorm():
"""Test batchnorm operator for Vitis-AI DPUCADX8G and DPUCZDX8G-zcu104
targets"""
@@ -323,7 +310,6 @@ if __name__ == "__main__":
test_conv2d()
test_depthwise_conv()
test_bias_add()
- test_relu()
test_add()
test_max_pool2d()
test_global_max_pool2d()
