From 5e0f6667f33abf4809f1349c9528108a58e25afd Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Tue, 11 Feb 2020 23:41:38 +0100
Subject: [PATCH] [Transform] set domain and backend correctly in
InsertTLastMarker
---
src/finn/transformation/fpgadataflow/insert_tlastmarker.py | 2 ++
tests/fpgadataflow/test_create_dataflow_partition.py | 1 +
2 files changed, 3 insertions(+)
diff --git a/src/finn/transformation/fpgadataflow/insert_tlastmarker.py b/src/finn/transformation/fpgadataflow/insert_tlastmarker.py
index d174f3acb..ed8b96852 100644
--- a/src/finn/transformation/fpgadataflow/insert_tlastmarker.py
+++ b/src/finn/transformation/fpgadataflow/insert_tlastmarker.py
@@ -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)
diff --git a/tests/fpgadataflow/test_create_dataflow_partition.py b/tests/fpgadataflow/test_create_dataflow_partition.py
index ed5b95654..af3dc5816 100644
--- a/tests/fpgadataflow/test_create_dataflow_partition.py
+++ b/tests/fpgadataflow/test_create_dataflow_partition.py
@@ -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
--
GitLab