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

[Transform] set domain and backend correctly in InsertTLastMarker

parent 2b13d42a
No related branches found
No related tags found
No related merge requests found
......@@ -39,6 +39,8 @@ class InsertTLastMarker(Transformation):
[graph_out_name],
NumIters=num_iters,
StreamWidth=stream_width,
domain="finn",
backend="fpgadataflow",
)
model.graph.node.append(tlast_node)
return (model, True)
......@@ -36,6 +36,7 @@ def test_dataflow_partition_tlastmarker():
model = ModelWrapper(model_path)
model = model.transform(InsertTLastMarker())
assert model.graph.node[-1].op_type == "TLastMarker"
assert model.graph.node[-1].domain == "finn"
tl_node = getCustomOp(model.graph.node[-1])
assert tl_node.get_nodeattr("NumIters") == 1
assert tl_node.get_nodeattr("StreamWidth") == 320
......
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