diff --git a/tests/fpgadataflow/test_fpgadataflow_globalaccpool.py b/tests/fpgadataflow/test_fpgadataflow_globalaccpool.py index bbfa153c6fd3959d07f8f1883fa88949072f3511..b46391daf629e97c24c2950aefad3cbc5055c345 100644 --- a/tests/fpgadataflow/test_fpgadataflow_globalaccpool.py +++ b/tests/fpgadataflow/test_fpgadataflow_globalaccpool.py @@ -63,7 +63,7 @@ def make_accpool_modelwrapper(ch, pe, idim, idt): numInputVectors=[1, idim, idim], ) graph = helper.make_graph( - nodes=[accpool_node], name="graph", inputs=[inp], outputs=[outp], + nodes=[accpool_node], name="graph", inputs=[inp], outputs=[outp] ) model = helper.make_model(graph, producer_name="thresholding-model") @@ -88,6 +88,7 @@ def prepare_inputs(input_tensor, idt): @pytest.mark.parametrize("imdim", [7]) # execution mode @pytest.mark.parametrize("exec_mode", ["cppsim", "rtlsim"]) +@pytest.mark.vivado def test_fpgadataflow_globalaccpool(idt, ch, fold, imdim, exec_mode): if fold == -1: pe = 1