From b2eca2212a777c5a0b4e9863fb052e15db0fffc5 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Tue, 24 May 2022 17:25:34 +0100
Subject: [PATCH] [Builder] Add signature parameter to build flow

---
 src/finn/builder/build_dataflow_config.py | 4 ++++
 src/finn/builder/build_dataflow_steps.py  | 1 +
 2 files changed, 5 insertions(+)

diff --git a/src/finn/builder/build_dataflow_config.py b/src/finn/builder/build_dataflow_config.py
index 6abac895a..1ef82a4e3 100644
--- a/src/finn/builder/build_dataflow_config.py
+++ b/src/finn/builder/build_dataflow_config.py
@@ -209,6 +209,10 @@ class DataflowBuildConfig:
     #: the full list of layer IP build directories. By default, synthesis will not run.
     stitched_ip_gen_dcp: Optional[bool] = False
 
+    #: (Optional) Insert a signature node to the stitched-IP to read/write information
+    #: to the design: e.g. Customer signature, application signature, version
+    signature: Optional[bool] = False
+
     #: (Optional) Control the maximum width of the per-PE MVAU stream while
     #: exploring the parallelization attributes to reach target_fps
     #: Only relevant if target_fps is specified.
diff --git a/src/finn/builder/build_dataflow_steps.py b/src/finn/builder/build_dataflow_steps.py
index 85f150dcf..852705c0d 100644
--- a/src/finn/builder/build_dataflow_steps.py
+++ b/src/finn/builder/build_dataflow_steps.py
@@ -511,6 +511,7 @@ def step_create_stitched_ip(model: ModelWrapper, cfg: DataflowBuildConfig):
                 cfg._resolve_fpga_part(),
                 cfg.synth_clk_period_ns,
                 vitis=cfg.stitched_ip_gen_dcp,
+                signature=cfg.signature,
             )
         )
         # TODO copy all ip sources into output dir? as zip?
-- 
GitLab