Skip to content
Snippets Groups Projects
Commit 0b58c6c9 authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Test] put generated models under /tmp/finn

parent cdc41832
No related branches found
No related tags found
No related merge requests found
......@@ -15,13 +15,14 @@ from finn.transformation.fpgadataflow.make_pynq_proj import MakePYNQProject
from finn.transformation.fpgadataflow.synth_pynq_proj import SynthPYNQProject
from finn.transformation.general import GiveUniqueNodeNames
# TODO control board/part for tests from a global place
# settings for Ultra96
test_fpga_part = "xczu3eg-sbva484-1-e"
test_pynq_board = "Ultra96"
# settings for PYNQ-Z1
test_fpga_part = "xc7z020clg400-1"
test_pynq_board = "Pynq-Z1"
#test_fpga_part = "xc7z020clg400-1"
#test_pynq_board = "Pynq-Z1"
def create_two_fc_model():
......@@ -138,12 +139,12 @@ def test_fpgadataflow_ipstitch_gen_two_fc_model():
assert model.graph.node[0].op_type == "StreamingFCLayer_Batch"
assert model.graph.node[1].op_type == "StreamingFCLayer_Batch"
assert model.graph.node[2].op_type == "TLastMarker"
model.save("/tmp/test_fpgadataflow_ip_gen_two_fc_model.onnx")
model.save("/tmp/finn/test_fpgadataflow_ip_gen_two_fc_model.onnx")
@pytest.mark.dependency(depends=["test_fpgadataflow_ipstitch_gen_two_fc_model"])
def test_fpgadataflow_ipstitch_do_stitch():
model = ModelWrapper("/tmp/test_fpgadataflow_ip_gen_two_fc_model.onnx")
model = ModelWrapper("/tmp/finn/test_fpgadataflow_ip_gen_two_fc_model.onnx")
model = model.transform(CodeGen_ipstitch(test_fpga_part))
vivado_stitch_proj_dir = model.get_metadata_prop("vivado_stitch_proj")
assert vivado_stitch_proj_dir is not None
......@@ -152,22 +153,22 @@ def test_fpgadataflow_ipstitch_do_stitch():
vivado_stitch_vlnv = model.get_metadata_prop("vivado_stitch_vlnv")
assert vivado_stitch_vlnv is not None
assert vivado_stitch_vlnv == "xilinx_finn:finn:finn_design:1.0"
model.save("/tmp/test_fpgadataflow_ip_stitch.onnx")
model.save("/tmp/finn/test_fpgadataflow_ip_stitch.onnx")
@pytest.mark.dependency(depends=["test_fpgadataflow_ipstitch_do_stitch"])
def test_fpgadataflow_ipstitch_pynq_projgen():
model = ModelWrapper("/tmp/test_fpgadataflow_ip_stitch.onnx")
model = ModelWrapper("/tmp/finn/test_fpgadataflow_ip_stitch.onnx")
model = model.transform(MakePYNQProject(test_pynq_board))
vivado_pynq_proj_dir = model.get_metadata_prop("vivado_pynq_proj")
assert vivado_pynq_proj_dir is not None
assert os.path.isdir(vivado_pynq_proj_dir)
model.save("/tmp/test_fpgadataflow_pynq_projgen.onnx")
model.save("/tmp/finn/test_fpgadataflow_pynq_projgen.onnx")
@pytest.mark.dependency(depends=["test_fpgadataflow_ipstitch_pynq_projgen"])
def test_fpgadataflow_ipstitch_pynq_synth():
model = ModelWrapper("/tmp/test_fpgadataflow_pynq_projgen.onnx")
model = ModelWrapper("/tmp/finn/test_fpgadataflow_pynq_projgen.onnx")
model = model.transform(SynthPYNQProject())
bitfile = model.get_metadata_prop("vivado_pynq_bitfile")
assert bitfile is not None
......
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