Skip to content

Commit a69e28e

Browse files
authored
Arm Backend: Remove unused param in aot_arm_compiler.py (#10970)
Signed-off-by: Martin Lindström <martin.lindstroem@arm.com>
1 parent d7699d6 commit a69e28e

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

examples/arm/aot_arm_compiler.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -673,9 +673,7 @@ def save_bpte_program(exec_prog, original_model: torch.nn.Module, output_name: s
673673
save_bundled_program(exec_prog, method_test_suites, output_name)
674674

675675

676-
def quantize_model(
677-
exported_program, args, model: torch.nn.Module, example_inputs, compile_spec
678-
):
676+
def quantize_model(args, model: torch.nn.Module, example_inputs, compile_spec):
679677
model_int8 = quantize(
680678
model,
681679
args.model_name,
@@ -707,7 +705,7 @@ def to_edge_TOSA_delegate(
707705
model_int8 = None
708706
if args.quantize:
709707
model_int8, exported_program = quantize_model(
710-
exported_program, args, model, example_inputs, compile_spec
708+
args, model, example_inputs, compile_spec
711709
)
712710
model = model_int8
713711

@@ -743,7 +741,7 @@ def to_edge_no_delegate(exported_program, args, model: torch.nn.Module, example_
743741
args.memory_mode,
744742
)
745743
model, exported_program = quantize_model(
746-
exported_program, args, model, example_inputs, compile_spec
744+
args, model, example_inputs, compile_spec
747745
)
748746
model_int8 = model
749747

0 commit comments

Comments
 (0)