Skip to content
Snippets Groups Projects
Commit c60c27fb authored by auphelia's avatar auphelia
Browse files

[Tests] Add missing marker and clean up topk test

parent 0d8896cf
No related branches found
No related tags found
No related merge requests found
......@@ -72,6 +72,7 @@ def make_concat_model(i_shapes, idt):
@pytest.mark.parametrize("exec_mode", ["cppsim", "rtlsim"])
@pytest.mark.parametrize("idt", [DataType["INT4"]])
@pytest.mark.fpgadataflow
@pytest.mark.vivado
@pytest.mark.slow
def test_fpgadataflow_concat(exec_mode, idt):
......@@ -107,6 +108,7 @@ def test_fpgadataflow_concat(exec_mode, idt):
assert (exp_out == ret_sim[oname]).all()
@pytest.mark.fpgadataflow
@pytest.mark.vivado
@pytest.mark.slow
def test_fpgadataflow_concat_stitchedip():
......
......@@ -47,7 +47,7 @@ from finn.transformation.streamline.reorder import MoveFlattenPastTopK
@pytest.mark.parametrize("data_layout", [DataLayout.NHWC, DataLayout.NCHW])
# batch size
@pytest.mark.parametrize("batch_size", [1, 2])
def test_move_flatten_past_affine(data_layout, batch_size):
def test_move_flatten_past_topk(data_layout, batch_size):
if data_layout == DataLayout.NHWC:
ishape = [batch_size, 1, 1, 1024]
oshape = [batch_size, 1024]
......
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