From ea776a1858680ee2baf4ba22064597217067c489 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Mon, 20 Apr 2020 17:22:45 +0100
Subject: [PATCH] [Test] Set pe, simd and fifo depth in tfc w1a1 test for
throughput test -> rename file
---
... test_end2end_tfc_w1a1_throughput_test.py} | 30 ++++++++++++++-----
1 file changed, 22 insertions(+), 8 deletions(-)
rename tests/end2end/{test_end2end_tfc_w1a1.py => test_end2end_tfc_w1a1_throughput_test.py} (94%)
diff --git a/tests/end2end/test_end2end_tfc_w1a1.py b/tests/end2end/test_end2end_tfc_w1a1_throughput_test.py
similarity index 94%
rename from tests/end2end/test_end2end_tfc_w1a1.py
rename to tests/end2end/test_end2end_tfc_w1a1_throughput_test.py
index 4fb278d18..946e84f2a 100644
--- a/tests/end2end/test_end2end_tfc_w1a1.py
+++ b/tests/end2end/test_end2end_tfc_w1a1_throughput_test.py
@@ -133,23 +133,37 @@ def test_end2end_tfc_w1a1_fold_and_tlastmarker():
fc1w = getCustomOp(fc_layers[1])
fc2w = getCustomOp(fc_layers[2])
fc3w = getCustomOp(fc_layers[3])
- fc0w.set_nodeattr("inFIFODepth", 50)
- fc0w.set_nodeattr("SIMD", 16)
+ fc0w.set_nodeattr("inFIFODepth", 256)
+ fc0w.set_nodeattr("SIMD", 196)
fc0w.set_nodeattr("PE", 16)
- fc0w.set_nodeattr("outFIFODepth", 4)
- fc1w.set_nodeattr("SIMD", 8)
- fc1w.set_nodeattr("PE", 8)
- fc1w.set_nodeattr("outFIFODepth", 4)
+ fc0w.set_nodeattr("outFIFODepth", 64)
+ fc1w.set_nodeattr("SIMD", 16)
+ fc1w.set_nodeattr("PE", 16)
+ fc1w.set_nodeattr("outFIFODepth", 64)
fc2w.set_nodeattr("SIMD", 16)
fc2w.set_nodeattr("PE", 16)
- fc2w.set_nodeattr("outFIFODepth", 4)
+ fc2w.set_nodeattr("outFIFODepth", 64)
fc3w.set_nodeattr("SIMD", 16)
fc3w.set_nodeattr("PE", 10)
- fc3w.set_nodeattr("outFIFODepth", 50)
+ fc3w.set_nodeattr("outFIFODepth", 10)
model = model.transform(InsertDWC())
model = model.transform(InsertFIFO())
model = model.transform(InsertTLastMarker())
model = model.transform(GiveUniqueNodeNames())
+ fifos = []
+ for n in model.graph.node:
+ if n.op_type == "StreamingFIFO":
+ fifos.append(n)
+ fifo0 = getCustomOp(fifos[0])
+ fifo1 = getCustomOp(fifos[1])
+ fifo2 = getCustomOp(fifos[2])
+ fifo3 = getCustomOp(fifos[3])
+ fifo4 = getCustomOp(fifos[4])
+ fifo0.set_nodeattr("depth", 256)
+ fifo1.set_nodeattr("depth", 64)
+ fifo2.set_nodeattr("depth", 64)
+ fifo3.set_nodeattr("depth", 64)
+ fifo4.set_nodeattr("depth", 10)
model = model.transform(AnnotateResources("estimate"))
model.save(build_dir + "/end2end_tfc_w1a1_folded.onnx")
--
GitLab