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

[Driver] add missing init.py files to generated folder, fix MB typo

parent a759dbaa
No related branches found
No related tags found
No related merge requests found
......@@ -439,13 +439,13 @@ class FINNExampleOverlay(Overlay):
total_in = 0
for i in range(self.num_inputs):
total_in += np.prod(self.ishape_packed(i))
res["DRAM_in_bandwidth[Mb/s]"] = total_in * 0.000001 / runtime
res["DRAM_in_bandwidth[MB/s]"] = total_in * 0.000001 / runtime
total_out = 0
for o in range(self.num_outputs):
total_out += np.prod(self.oshape_packed(o))
res["DRAM_out_bandwidth[Mb/s]"] = total_out * 0.000001 / runtime
res["DRAM_out_bandwidth[MB/s]"] = total_out * 0.000001 / runtime
for iwdma, iwbuf, iwdma_name in self.external_weights:
res["DRAM_extw_%s_bandwidth[Mb/s]" % iwdma_name] = (
res["DRAM_extw_%s_bandwidth[MB/s]" % iwdma_name] = (
self.batch_size * np.prod(iwbuf.shape) * 0.000001 / runtime
)
if self.platform == "zynq-iodma":
......
......@@ -118,12 +118,21 @@ class MakePYNQDriver(Transformation):
files_to_copy.append(
(qonnx_path + "/util/basic.py", qonnx_target_path + "/util/basic.py")
)
files_to_copy.append(
(qonnx_path + "/util/__init__.py", qonnx_target_path + "/util/__init__.py")
)
files_to_copy.append(
(
finn_util_path + "/data_packing.py",
finn_target_path + "/util/data_packing.py",
)
)
files_to_copy.append(
(
finn_util_path + "/__init__.py",
finn_target_path + "/util/__init__.py",
)
)
for (src_file, target_file) in files_to_copy:
shutil.copy(src_file, target_file)
# extract input-output shapes from the graph
......
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