From 23ad4eb7c8837941e317e2fab7b1e107fee2c04b Mon Sep 17 00:00:00 2001 From: Zingo Andersen Date: Wed, 26 Feb 2025 14:32:49 +0100 Subject: [PATCH] Arm backend: Fix for non quantized TOSA delegation in aot_arm_compiler Signed-off-by: Zingo Andersen Change-Id: I1581c1e8d578418105cdfc3b269fbbd84b0ea641 --- examples/arm/aot_arm_compiler.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/arm/aot_arm_compiler.py b/examples/arm/aot_arm_compiler.py index 1f224983d4e..5fb12342a2d 100644 --- a/examples/arm/aot_arm_compiler.py +++ b/examples/arm/aot_arm_compiler.py @@ -624,6 +624,7 @@ def save_bpte_program(exec_prog, original_model: torch.nn.Module, output_name: s def to_edge_TOSA_delegate( + exported_program, args, model: torch.nn.Module, ): @@ -687,7 +688,7 @@ def to_edge_TOSA_delegate( # Quantize if required model_int8 = None if args.delegate: - model_int8, edge = to_edge_TOSA_delegate(args, model) + model_int8, edge = to_edge_TOSA_delegate(exported_program, args, model) else: edge = to_edge_transform_and_lower( exported_program,