diff --git a/src/finn/builder/build_dataflow_config.py b/src/finn/builder/build_dataflow_config.py index e48aa4043864124cb6949f713ec13c5e762eb4ef..49f800f62a6f9c646c5223504d09cb90d3d144f7 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 b3f80e1bb5ae6427d8de2c6ca6b54c8180c34fe5..2216fcb3535360b6da5a43942bc46e81780fa341 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