From d1e9ea9f35e8a4645cd298efcabbe4942f31c9f3 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Thu, 23 Apr 2020 16:43:27 +0100
Subject: [PATCH] [Trafo-templates] Change driver.py to write network metrics
as dictionary to txt file
---
src/finn/transformation/fpgadataflow/templates.py | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/src/finn/transformation/fpgadataflow/templates.py b/src/finn/transformation/fpgadataflow/templates.py
index bc9372855..20b1bd58e 100644
--- a/src/finn/transformation/fpgadataflow/templates.py
+++ b/src/finn/transformation/fpgadataflow/templates.py
@@ -151,6 +151,7 @@ if __name__ == "__main__":
if exec_mode == "remote_pynq":
N = 1
elif exec_mode == "throughput_test":
+ res={}
N = 1000
else:
raise Exception("Exec mode has to be set to remote_pynq or throughput_test")
@@ -191,8 +192,10 @@ if __name__ == "__main__":
if exec_mode == "throughput_test":
end = time.time()
runtime = end - start
- file = open("nw_runtime.txt", "w")
- file.write(str(runtime))
+ res["runtime[ms]"] = runtime*1000
+ res["throughput[images/s]"] = N / runtime
+ file = open("nw_metrics.txt", "w")
+ file.write(str(res))
file.close()
else:
--
GitLab