From 5593117a2e6ceb0ea43d0aa628131de9f074afe4 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Thu, 8 Oct 2020 16:28:36 +0200
Subject: [PATCH] [Test] check outputs from fifo depth setting in end2end test
---
tests/end2end/test_end2end_bnn_pynq.py | 8 ++++++++
1 file changed, 8 insertions(+)
diff --git a/tests/end2end/test_end2end_bnn_pynq.py b/tests/end2end/test_end2end_bnn_pynq.py
index c81e3f9e6..84099339f 100644
--- a/tests/end2end/test_end2end_bnn_pynq.py
+++ b/tests/end2end/test_end2end_bnn_pynq.py
@@ -455,6 +455,14 @@ class TestEnd2End:
model = load_test_checkpoint_or_skip(prev_chkpt_name)
test_fpga_part = get_build_env(kind, target_clk_ns)["part"]
model = model.transform(InsertAndSetFIFODepths(test_fpga_part, target_clk_ns))
+ fifo_layers = model.get_nodes_by_op_type("StreamingFIFO")
+ assert len(fifo_layers) > 0
+ hls_layers = model.get_finn_nodes()
+ for node in hls_layers:
+ if node.op_type != "StreamingFIFO":
+ op_inst = getCustomOp(node)
+ assert op_inst.get_nodeattr("inFIFODepth") == 0
+ assert op_inst.get_nodeattr("outFIFODepth") == 0
model.save(get_checkpoint_name(topology, wbits, abits, "fifodepth_" + kind))
@pytest.mark.slow
--
GitLab