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

[Test] specify ram_style for 2 layers in end2end_tfc_w1a1

parent ff0f873d
No related branches found
No related tags found
No related merge requests found
......@@ -137,6 +137,7 @@ def test_end2end_tfc_w1a1_fold_and_tlastmarker():
fc0w.set_nodeattr("SIMD", 16)
fc0w.set_nodeattr("PE", 16)
fc0w.set_nodeattr("outFIFODepth", 4)
fc0w.set_nodeattr("ram_style", "block")
fc1w.set_nodeattr("SIMD", 8)
fc1w.set_nodeattr("PE", 8)
fc1w.set_nodeattr("outFIFODepth", 4)
......@@ -146,6 +147,7 @@ def test_end2end_tfc_w1a1_fold_and_tlastmarker():
fc3w.set_nodeattr("SIMD", 16)
fc3w.set_nodeattr("PE", 10)
fc3w.set_nodeattr("outFIFODepth", 50)
fc3w.set_nodeattr("ram_style", "distributed")
model = model.transform(InsertDWC())
model = model.transform(InsertTLastMarker())
model = model.transform(GiveUniqueNodeNames())
......
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