From dc84ba7857ef50d18144c52b8ab1872726e945ea Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Tue, 11 Aug 2020 13:06:09 +0200
Subject: [PATCH] [Test] don't use debug mode for ZynqBuild end2end tests
---
tests/end2end/test_zynqbuild_end2end_cnv_w1a1.py | 4 +---
tests/end2end/test_zynqbuild_end2end_tfc_w1a1.py | 4 +---
tests/end2end/test_zynqbuild_end2end_tfc_w2a2.py | 4 +---
3 files changed, 3 insertions(+), 9 deletions(-)
diff --git a/tests/end2end/test_zynqbuild_end2end_cnv_w1a1.py b/tests/end2end/test_zynqbuild_end2end_cnv_w1a1.py
index 38366f5bd..1a1b21afa 100644
--- a/tests/end2end/test_zynqbuild_end2end_cnv_w1a1.py
+++ b/tests/end2end/test_zynqbuild_end2end_cnv_w1a1.py
@@ -182,9 +182,7 @@ def test_end2end_zynqbuild_cnv_w1a1_build():
model = load_test_checkpoint_or_skip(
build_dir + "/end2end_zynqbuild_cnv_w1a1_pynq_driver.onnx"
)
- model = model.transform(
- ZynqBuild(test_pynq_board, target_clk_ns, enable_debug=True)
- )
+ model = model.transform(ZynqBuild(test_pynq_board, target_clk_ns))
model = model.transform(AnnotateResources("synth"))
model.save(build_dir + "/end2end_zynqbuild_cnv_w1a1_build.onnx")
diff --git a/tests/end2end/test_zynqbuild_end2end_tfc_w1a1.py b/tests/end2end/test_zynqbuild_end2end_tfc_w1a1.py
index 4a5901d32..98ccc9318 100644
--- a/tests/end2end/test_zynqbuild_end2end_tfc_w1a1.py
+++ b/tests/end2end/test_zynqbuild_end2end_tfc_w1a1.py
@@ -166,9 +166,7 @@ def test_end2end_zynqbuild_tfc_w1a1_build():
model = load_test_checkpoint_or_skip(
build_dir + "/end2end_zynqbuild_tfc_w1a1_pynq_driver.onnx"
)
- model = model.transform(
- ZynqBuild(test_pynq_board, target_clk_ns, enable_debug=True)
- )
+ model = model.transform(ZynqBuild(test_pynq_board, target_clk_ns))
model = model.transform(AnnotateResources("synth"))
model.save(build_dir + "/end2end_zynqbuild_tfc_w1a1_build.onnx")
diff --git a/tests/end2end/test_zynqbuild_end2end_tfc_w2a2.py b/tests/end2end/test_zynqbuild_end2end_tfc_w2a2.py
index 4e96635eb..b8db8c1a4 100644
--- a/tests/end2end/test_zynqbuild_end2end_tfc_w2a2.py
+++ b/tests/end2end/test_zynqbuild_end2end_tfc_w2a2.py
@@ -156,9 +156,7 @@ def test_end2end_zynqbuild_tfc_w2a2_build():
model = load_test_checkpoint_or_skip(
build_dir + "/end2end_zynqbuild_tfc_w2a2_pynq_driver.onnx"
)
- model = model.transform(
- ZynqBuild(test_pynq_board, target_clk_ns, enable_debug=True)
- )
+ model = model.transform(ZynqBuild(test_pynq_board, target_clk_ns))
model = model.transform(AnnotateResources("synth"))
model.save(build_dir + "/end2end_zynqbuild_tfc_w2a2_build.onnx")
--
GitLab