Arm backend: Cherry-pick tosa patch to speed up large models (#15592)

* Only affects models >2gb

Signed-off-by: Ryan O'Shea <ryan.oshea3@arm.com>

Author: ArmRyan

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

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