From 906bb6ae11e9641963b00e8ada1f61dbaf8a3c09 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Tue, 18 May 2021 15:07:58 +0100
Subject: [PATCH] [Test] only do lfc-1-1 as part of bnn-pynq end2end tests
---
tests/end2end/test_end2end_bnn_pynq.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tests/end2end/test_end2end_bnn_pynq.py b/tests/end2end/test_end2end_bnn_pynq.py
index 6d0e028a1..d0a9aa1c0 100644
--- a/tests/end2end/test_end2end_bnn_pynq.py
+++ b/tests/end2end/test_end2end_bnn_pynq.py
@@ -307,8 +307,8 @@ class TestEnd2End:
def test_export(self, topology, wbits, abits):
if wbits > abits:
pytest.skip("No wbits > abits end2end network configs for now")
- if topology == "lfc" and wbits > 1:
- pytest.skip("Skipping non-existing lfc configs")
+ if topology == "lfc" and not (wbits == 1 and abits == 1):
+ pytest.skip("Skipping certain lfc configs")
(model, ishape) = get_trained_network_and_ishape(topology, wbits, abits)
chkpt_name = get_checkpoint_name(topology, wbits, abits, "export")
bo.export_finn_onnx(model, ishape, chkpt_name)
--
GitLab