From 9650adc1da3fcf21eab34118bc6f9324043603f3 Mon Sep 17 00:00:00 2001 From: Ryan O'Shea Date: Tue, 4 Nov 2025 12:27:23 +0100 Subject: [PATCH] Arm backend: Cherry-pick tosa patch to speed up large models * Only affects models >2gb Signed-off-by: Ryan O'Shea Change-Id: Id5ec79eecc4f1ae8d864e35ed13b34bbf761d30b --- backends/arm/tosa/backend.py | 9 ++++++++- examples/arm/setup.sh | 2 ++ 2 files changed, 10 insertions(+), 1 deletion(-) diff --git a/backends/arm/tosa/backend.py b/backends/arm/tosa/backend.py index 42af22ff09f..1631526e360 100644 --- a/backends/arm/tosa/backend.py +++ b/backends/arm/tosa/backend.py @@ -133,7 +133,14 @@ def _preprocess( # noqa: C901 if not artifact_path: artifact_path = "" - tosa_graph = ts.TosaSerializer(artifact_path) + version = tosa_spec.version + tosa_graph = ts.TosaSerializer( + artifact_path, + targetMajor=version.major, + targetMinor=version.minor, + targetPatch=version.micro, + targetDraft=False, + ) if not ( tosa_spec.version.major == ts.TOSA_VERSION_MAJOR diff --git a/examples/arm/setup.sh b/examples/arm/setup.sh index db32c0c416f..a11b4a2eebd 100755 --- a/examples/arm/setup.sh +++ b/examples/arm/setup.sh @@ -290,7 +290,9 @@ if [[ $is_script_sourced -eq 0 ]]; then fi pushd tosa-tools + git fetch origin main git checkout 8468d041c50c6d806f3c1c18c66d7ef641e46580 # serialization lib pybindings + git cherry-pick 368f0cd745b2a1569bf36f077daeba95775de192 # perf fix for >2gb models if [[ ! -d "reference_model" ]]; then log_step "main" "[error] Missing reference_model directory in tosa-tools repo." exit 1