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): ...@@ -439,13 +439,13 @@ class FINNExampleOverlay(Overlay):
total_in = 0 total_in = 0
for i in range(self.num_inputs): for i in range(self.num_inputs):
total_in += np.prod(self.ishape_packed(i)) 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 total_out = 0
for o in range(self.num_outputs): for o in range(self.num_outputs):
total_out += np.prod(self.oshape_packed(o)) 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: 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 self.batch_size * np.prod(iwbuf.shape) * 0.000001 / runtime
) )
if self.platform == "zynq-iodma": if self.platform == "zynq-iodma":
......
...@@ -118,12 +118,21 @@ class MakePYNQDriver(Transformation): ...@@ -118,12 +118,21 @@ class MakePYNQDriver(Transformation):
files_to_copy.append( files_to_copy.append(
(qonnx_path + "/util/basic.py", qonnx_target_path + "/util/basic.py") (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( files_to_copy.append(
( (
finn_util_path + "/data_packing.py", finn_util_path + "/data_packing.py",
finn_target_path + "/util/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: for (src_file, target_file) in files_to_copy:
shutil.copy(src_file, target_file) shutil.copy(src_file, target_file)
# extract input-output shapes from the graph # 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