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

[PYNQ] add N and fclk_mhz to generated driver stats

parent abf46c13
No related branches found
No related tags found
No related merge requests found
......@@ -101,6 +101,7 @@ from finn.util.data_packing import (
packed_bytearray_to_finnpy
)
from finn.core.datatype import DataType
from pynq.ps import Clocks
class FINNAccelDriver():
def __init__(self, N, bitfile):
......@@ -228,6 +229,8 @@ if __name__ == "__main__":
res["throughput[images/s]"] = N / runtime
res["DRAM_in_bandwidth[Mb/s]"] = np.prod(finnDriver.ishape_packed)*0.000001 / runtime
res["DRAM_out_bandwidth[Mb/s]"] = np.prod(finnDriver.oshape_packed)*0.000001 / runtime
res["fclk[mhz]"] = Clocks.fclk0_mhz
res["N"] = N
file = open("nw_metrics.txt", "w")
file.write(str(res))
file.close()
......
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