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

[Refactor] use metadata_prop accessors from modelwrapper

parent 7c7a114a
No related branches found
No related tags found
No related merge requests found
......@@ -2,8 +2,6 @@ import os
import subprocess
import tempfile as tmp
import onnx
from finn.core.utils import get_by_name
from finn.transformation import Transformation
......@@ -91,16 +89,8 @@ class CodeGen_ipstitch(Transformation):
)
# create a temporary folder for the project
vivado_proj = get_by_name(model.model.metadata_props, "vivado_proj", "key")
if vivado_proj is None or not os.path.isdir(vivado_proj.value):
vivado_proj = onnx.StringStringEntryProto()
vivado_proj.key = "vivado_proj"
vivado_proj.value = tmp.mkdtemp(prefix="vivado_proj_")
model.model.metadata_props.append(vivado_proj)
else:
# create new temporary folder, don't reuse old one
vivado_proj.value = tmp.mkdtemp(prefix="vivado_proj_")
vivado_proj_dir = vivado_proj.value
vivado_proj_dir = tmp.mkdtemp(prefix="vivado_proj_")
model.set_metadata_prop("vivado_proj", vivado_proj_dir)
# start building the tcl script
tcl = []
# create vivado project
......
......@@ -3,7 +3,6 @@ import os.path
import numpy as np
from onnx import TensorProto, helper
import finn.core.utils.get_by_name as get_by_name
from finn.core.datatype import DataType
from finn.core.modelwrapper import ModelWrapper
from finn.core.utils import calculate_signed_dot_prod_range, gen_finn_dt_tensor
......@@ -113,9 +112,8 @@ def test_fpgadataflow_ip_stitch():
model = model.transform(CodeGen_ipgen("xc7z020clg400-1", 5))
model = model.transform(HLSSynth_IPGen())
model = model.transform(CodeGen_ipstitch("xc7z020clg400-1"))
vivado_proj = get_by_name(model.model.metadata_props, "vivado_proj", "key")
assert vivado_proj is not None
vivado_proj_dir = vivado_proj.value
vivado_proj_dir = model.get_metadata_prop("vivado_proj")
assert vivado_proj_dir is not None
assert os.path.isdir(vivado_proj_dir)
assert os.path.isfile(vivado_proj_dir + "/ip/component.xml")
model = model.transform(CleanUp())
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