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

[Test] reflect TLastMarker changes for i/o shape fxns

parent 31be47b1
No related branches found
No related tags found
No related merge requests found
......@@ -40,6 +40,7 @@ def test_dataflow_partition_tlastmarker():
tl_node = getCustomOp(model.graph.node[-1])
assert tl_node.get_nodeattr("NumIters") == 1
assert tl_node.get_nodeattr("StreamWidth") == 320
assert tl_node.get_nodeattr("ElemWidth") == 32
model.save(build_dir + "/test_dataflow_partition_tlastmarker.onnx")
model = model.transform(InsertTLastMarker())
model.save(build_dir + "/test_dataflow_partition_tlastmarker2.onnx")
......@@ -75,6 +75,7 @@ def create_one_fc_model():
domain="finn",
backend="fpgadataflow",
NumIters=1,
ElemWidth=odt.bitwidth(),
StreamWidth=odt.bitwidth() * m,
)
......@@ -164,6 +165,7 @@ def create_two_fc_model():
backend="fpgadataflow",
NumIters=m,
StreamWidth=2,
ElemWidth=odt.bitwidth(),
)
graph = helper.make_graph(
......
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