From b70659cbf60ca58341043dedf84f3b8eb1526e19 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Tue, 19 Jul 2022 19:38:05 +0200
Subject: [PATCH] [Build] New flag "verbose" to control build_dataflow
verbosity
---
src/finn/builder/build_dataflow.py | 14 ++++++++------
src/finn/builder/build_dataflow_config.py | 4 ++++
2 files changed, 12 insertions(+), 6 deletions(-)
diff --git a/src/finn/builder/build_dataflow.py b/src/finn/builder/build_dataflow.py
index 238083f65..d6864994a 100644
--- a/src/finn/builder/build_dataflow.py
+++ b/src/finn/builder/build_dataflow.py
@@ -155,12 +155,14 @@ def build_dataflow_cfg(model_filename, cfg: DataflowBuildConfig):
% (step_name, step_num, len(build_dataflow_steps))
)
# redirect output to logfile
- sys.stdout = stdout_logger
- sys.stderr = stderr_logger
- print(
- "Running step: %s [%d/%d]"
- % (step_name, step_num, len(build_dataflow_steps))
- )
+ if not cfg.verbose:
+ sys.stdout = stdout_logger
+ sys.stderr = stderr_logger
+ # also log current step name to logfile
+ print(
+ "Running step: %s [%d/%d]"
+ % (step_name, step_num, len(build_dataflow_steps))
+ )
# run the step
step_start = time.time()
model = transform_step(model, cfg)
diff --git a/src/finn/builder/build_dataflow_config.py b/src/finn/builder/build_dataflow_config.py
index 09e9ec3a5..92263bd82 100644
--- a/src/finn/builder/build_dataflow_config.py
+++ b/src/finn/builder/build_dataflow_config.py
@@ -285,6 +285,10 @@ class DataflowBuildConfig:
#: Whether pdb postmortem debuggig will be launched when the build fails
enable_build_pdb_debug: Optional[bool] = True
+ #: When True, all warnings and compiler output will be printed in stdout.
+ #: Otherwise, these will be suppressed and only appear in the build log.
+ verbose: Optional[bool] = False
+
#: If given, only run the steps in the list. If not, run default steps.
#: See `default_build_dataflow_steps` for the default list of steps.
#: When specified:
--
GitLab