Skip to content

Commit 8210d02

Browse files
Arm backend: Fix numerical accuracy error. (#13578)
A missing .graph caused numerical errors to show up as "AttributeError: 'GraphModule' object has no attribute 'output_node'" Signed-off-by: Adrian Lundell <[email protected]>
1 parent 39054a2 commit 8210d02

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

backends/arm/test/tester/analyze_output_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -253,7 +253,7 @@ def dump_error_output(
253253
export_stage = tester.stages.get(StageType.EXPORT, None)
254254
quantize_stage = tester.stages.get(StageType.QUANTIZE, None)
255255
if export_stage is not None and quantize_stage is not None:
256-
output_node = export_stage.artifact.graph_module.output_node()
256+
output_node = export_stage.artifact.graph_module.graph.output_node()
257257
qp_input = get_input_quantization_params(export_stage.artifact)
258258
qp_output = get_output_quantization_params(output_node)
259259
logger.error(f"Input QuantArgs: {qp_input}")

0 commit comments

Comments
 (0)