diff --git a/src/finn/transformation/fpgadataflow/templates.py b/src/finn/transformation/fpgadataflow/templates.py index bc9372855d4bddd3012a1befab66f601bcf5708b..20b1bd58eb022ac7e068a501d37d55ecdf8082af 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: