Skip to content

Commit 189ffaa

Browse files
Arm backend: Fix deepcopy & require grad issues. (#20114)
Change-Id: I7a813807c0e7a2734a96c317d2f198ae489de285 cc @digantdesai @freddan80 @per @zingo @oscarandersson8218 @mansnils @Sebastian-Larsson @robell @rascani Signed-off-by: Michiel Olieslagers <michiel.olieslagers@arm.com>
1 parent 951cd2e commit 189ffaa

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

backends/arm/scripts/aot_arm_compiler.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1034,6 +1034,7 @@ def main() -> None: # noqa: C901
10341034
args.calibration_data, example_inputs
10351035
)
10361036
model = original_model.eval()
1037+
model.requires_grad_(False)
10371038

10381039
# export under the assumption we quantize, the exported form also works
10391040
# in to_edge if we don't quantize
@@ -1115,8 +1116,6 @@ def main() -> None: # noqa: C901
11151116

11161117
dump_delegation_info(edge, args.intermediates)
11171118

1118-
edge_program_manager_copy = copy.deepcopy(edge)
1119-
11201119
try:
11211120
exec_prog = edge.to_executorch(
11221121
config=ExecutorchBackendConfig(extract_delegate_segments=False)
@@ -1175,6 +1174,7 @@ def main() -> None: # noqa: C901
11751174
if args.bundleio or args.etrecord:
11761175
etrecord_file_name = os.path.splitext(output_file_name)[0] + "_etrecord.bin"
11771176
try:
1177+
edge_program_manager_copy = copy.deepcopy(edge)
11781178
generate_etrecord(etrecord_file_name, edge_program_manager_copy, exec_prog)
11791179
print(f"ETRecord saved as {etrecord_file_name}")
11801180
except Exception as e:

0 commit comments

Comments
 (0)