Skip to content

Commit df9e58b

Browse files
Martin LindströmMartin Lindström
authored andcommitted
Arm Backend: Remove unused param in aot_arm_compiler.py
Signed-off-by: Martin Lindström <[email protected]> Change-Id: I756f7daf44097dcb9c91a4fdf27cacf979a2e871
1 parent d4906e2 commit df9e58b

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
@@ -671,9 +671,7 @@ def save_bpte_program(exec_prog, original_model: torch.nn.Module, output_name: s
671671
save_bundled_program(exec_prog, method_test_suites, output_name)
672672

673673

674-
def quantize_model(
675-
exported_program, args, model: torch.nn.Module, example_inputs, compile_spec
676-
):
674+
def quantize_model(args, model: torch.nn.Module, example_inputs, compile_spec):
677675
model_int8 = quantize(
678676
model,
679677
args.model_name,
@@ -705,7 +703,7 @@ def to_edge_TOSA_delegate(
705703
model_int8 = None
706704
if args.quantize:
707705
model_int8, exported_program = quantize_model(
708-
exported_program, args, model, example_inputs, compile_spec
706+
args, model, example_inputs, compile_spec
709707
)
710708
model = model_int8
711709

@@ -741,7 +739,7 @@ def to_edge_no_delegate(exported_program, args, model: torch.nn.Module, example_
741739
args.memory_mode,
742740
)
743741
model, exported_program = quantize_model(
744-
exported_program, args, model, example_inputs, compile_spec
742+
args, model, example_inputs, compile_spec
745743
)
746744
model_int8 = model
747745

0 commit comments

Comments
 (0)