diff --git a/tests/end2end/test_end2end_bnn_pynq.py b/tests/end2end/test_end2end_bnn_pynq.py index c81e3f9e6e80271ca7f927fe97c9d6fbe02e81ea..84099339f9ee978e11353738ebecff7e58c0cf1d 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