From b091e1e8f9f75ca7b641a512f591fb5f701fbf1f Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Tue, 15 Sep 2020 23:20:10 +0200
Subject: [PATCH] [Test] make end2end rtlsim top-1 measurement optional on
env.var.
---
tests/end2end/test_end2end_bnn_pynq.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/tests/end2end/test_end2end_bnn_pynq.py b/tests/end2end/test_end2end_bnn_pynq.py
index ce5a8e84a..398aa90b5 100644
--- a/tests/end2end/test_end2end_bnn_pynq.py
+++ b/tests/end2end/test_end2end_bnn_pynq.py
@@ -397,6 +397,8 @@ class TestEnd2End:
@pytest.mark.slow
@pytest.mark.parametrize("kind", ["zynq"])
def test_rtlsim_top1(self, topology, wbits, abits, kind):
+ if "TEST_RTLSIM_TOP1" not in os.environ:
+ pytest.skip("TEST_RTLSIM_TOP1 not set")
if "fc" not in topology:
pytest.skip("Top-1 rtlsim test currently for MNIST only")
rtlsim_chkpt = get_checkpoint_name(
--
GitLab