Skip to content
Snippets Groups Projects
Commit 906bb6ae authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Test] only do lfc-1-1 as part of bnn-pynq end2end tests

parent 1943d4c9
No related branches found
No related tags found
No related merge requests found
...@@ -307,8 +307,8 @@ class TestEnd2End: ...@@ -307,8 +307,8 @@ class TestEnd2End:
def test_export(self, topology, wbits, abits): def test_export(self, topology, wbits, abits):
if wbits > abits: if wbits > abits:
pytest.skip("No wbits > abits end2end network configs for now") pytest.skip("No wbits > abits end2end network configs for now")
if topology == "lfc" and wbits > 1: if topology == "lfc" and not (wbits == 1 and abits == 1):
pytest.skip("Skipping non-existing lfc configs") pytest.skip("Skipping certain lfc configs")
(model, ishape) = get_trained_network_and_ishape(topology, wbits, abits) (model, ishape) = get_trained_network_and_ishape(topology, wbits, abits)
chkpt_name = get_checkpoint_name(topology, wbits, abits, "export") chkpt_name = get_checkpoint_name(topology, wbits, abits, "export")
bo.export_finn_onnx(model, ishape, chkpt_name) bo.export_finn_onnx(model, ishape, chkpt_name)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment