From 10f60496bf41e73c1a8f731d568f94130ed684a4 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Tue, 12 Jul 2022 13:00:20 +0200
Subject: [PATCH] [Driver] add missing init.py files to generated folder, fix
MB typo
---
src/finn/qnn-data/templates/driver/driver_base.py | 6 +++---
src/finn/transformation/fpgadataflow/make_pynq_driver.py | 9 +++++++++
2 files changed, 12 insertions(+), 3 deletions(-)
diff --git a/src/finn/qnn-data/templates/driver/driver_base.py b/src/finn/qnn-data/templates/driver/driver_base.py
index 497477da9..209676058 100644
--- a/src/finn/qnn-data/templates/driver/driver_base.py
+++ b/src/finn/qnn-data/templates/driver/driver_base.py
@@ -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":
diff --git a/src/finn/transformation/fpgadataflow/make_pynq_driver.py b/src/finn/transformation/fpgadataflow/make_pynq_driver.py
index 863523605..dce98e54a 100644
--- a/src/finn/transformation/fpgadataflow/make_pynq_driver.py
+++ b/src/finn/transformation/fpgadataflow/make_pynq_driver.py
@@ -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
--
GitLab