Skip to content
Snippets Groups Projects
Commit 2a73e420 authored by Hendrik Borras's avatar Hendrik Borras
Browse files

Enabled FINNManager export for test_end2end_cybsec_mlp_build.

parent 866bc31e
No related branches found
No related tags found
No related merge requests found
...@@ -155,7 +155,7 @@ def test_end2end_cybsec_mlp_export(QONNX_export): ...@@ -155,7 +155,7 @@ def test_end2end_cybsec_mlp_export(QONNX_export):
assert tuple(finn_model.get_tensor_shape(finnonnx_in_tensor_name)) == (1, 600) assert tuple(finn_model.get_tensor_shape(finnonnx_in_tensor_name)) == (1, 600)
# verify a few exported ops # verify a few exported ops
if QONNX_export: if QONNX_export:
# The first "Mul" node dosen't exist in the QONNX export, # The first "Mul" node doesn't exist in the QONNX export,
# because the QuantTensor scale is not exported. # because the QuantTensor scale is not exported.
# However, this node would have been unity scale anyways and # However, this node would have been unity scale anyways and
# the models are still equivalent. # the models are still equivalent.
...@@ -180,8 +180,7 @@ def test_end2end_cybsec_mlp_export(QONNX_export): ...@@ -180,8 +180,7 @@ def test_end2end_cybsec_mlp_export(QONNX_export):
@pytest.mark.slow @pytest.mark.slow
@pytest.mark.vivado @pytest.mark.vivado
# @pytest.mark.parametrize("QONNX_export", [False, True]) @pytest.mark.parametrize("QONNX_export", [False, True])
@pytest.mark.parametrize("QONNX_export", [True])
def test_end2end_cybsec_mlp_build(QONNX_export): def test_end2end_cybsec_mlp_build(QONNX_export):
model_file = get_checkpoint_name("export", QONNX_export) model_file = get_checkpoint_name("export", QONNX_export)
load_test_checkpoint_or_skip(model_file) load_test_checkpoint_or_skip(model_file)
......
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