Skip to content
Snippets Groups Projects
Commit 3b999437 authored by Hendrik Borras's avatar Hendrik Borras
Browse files

Renamed QONNX to FINN verification step to QONNX_TO_FINN_PYTHON.

parent 222e7889
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment