From 3b9994378e91d8425f179f140f7db4c500eca632 Mon Sep 17 00:00:00 2001
From: Hendrik Borras <hendrikborras@web.de>
Date: Tue, 12 Oct 2021 14:14:55 +0100
Subject: [PATCH] Renamed QONNX to FINN verification step to
QONNX_TO_FINN_PYTHON.
---
src/finn/builder/build_dataflow_config.py | 2 +-
src/finn/builder/build_dataflow_steps.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/finn/builder/build_dataflow_config.py b/src/finn/builder/build_dataflow_config.py
index e48aa4043..49f800f62 100644
--- a/src/finn/builder/build_dataflow_config.py
+++ b/src/finn/builder/build_dataflow_config.py
@@ -90,7 +90,7 @@ class VerificationStepType(str, Enum):
"Steps at which FINN ONNX execution can be launched for verification."
#: verify after step_qonnx_to_finn, using Python execution
- FINN_ONNX_PYTHON = "finn_onnx_python"
+ QONNX_TO_FINN_PYTHON = "finn_onnx_python"
#: verify after step_tidy_up, using Python execution
TIDY_UP_PYTHON = "initial_python"
#: verify after step_streamline , using Python execution
diff --git a/src/finn/builder/build_dataflow_steps.py b/src/finn/builder/build_dataflow_steps.py
index b3f80e1bb..2216fcb35 100644
--- a/src/finn/builder/build_dataflow_steps.py
+++ b/src/finn/builder/build_dataflow_steps.py
@@ -208,7 +208,7 @@ def step_qonnx_to_finn(model: ModelWrapper, cfg: DataflowBuildConfig):
# QONNX to FINN-ONNX
model = model.transform(ConvertQONNXtoFINN())
- if VerificationStepType.FINN_ONNX_PYTHON in cfg._resolve_verification_steps():
+ if VerificationStepType.QONNX_TO_FINN_PYTHON in cfg._resolve_verification_steps():
verify_step(model, cfg, "initial_python", need_parent=False)
return model
--
GitLab