[Cria][Lllama runner] Use caching temp allocator

[kimishpatel]

Stack from ghstack (oldest at bottom):

• [Executorch] make slice_copy parallel #15830
• -> [Cria][Lllama runner] Use caching temp allocator #16080
• [Executorch][LLM] Use caching allocator for runner #15730
• [Executorch] Make module constructors uniform across #15729
• [Executorch] Use temp allocator for allocating scratch memory #15728
• [Executorch] Introduce caching cpu memory allocator #15611
• Reduce allocation overhead in quantized sdpa #15610

Use of caching allocator improves TITO model performance by 6+ %.

Will add repro instructions here but requires next diff to see the impact

Differential Revision: D85532078

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/16080

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

❌ 3 New Failures

As of commit c6289da with merge base 8af8252 ():

NEW FAILURES - The following jobs have failed:

• pull / android / build-llm-demo / linux-job (gh)
  /pytorch/executorch/extension/memory_allocator/cpu_caching_malloc_allocator.cpp:42:17: error: excess elements in scalar initializer
• pull / test-llama-runner-linux-android / linux-job (gh)
  /pytorch/executorch/extension/memory_allocator/cpu_caching_malloc_allocator.cpp:42:17: error: excess elements in scalar initializer
• pull / test-multimodal-linux (gemma3-4b) / linux-job (gh)
  RuntimeError: Command docker exec -t 6c45b33ca7fde2231af9e719c3b192b254b570a2f67fafcf6f01f42ddc5ab052 /exec failed with exit code 139

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[github-actions]

This PR needs a release notes: label

If your change should be included in the release notes (i.e. would users of this library care about this change?), please use a label starting with release notes:. This helps us keep track and include your important work in the next release notes.

To add a label, you can comment to pytorchbot, for example
@pytorchbot label "release notes: none"

For more information, see
https://github.com/pytorch/pytorch/wiki/PyTorch-AutoLabel-Bot#why-categorize-for-release-notes-and-how-does-it-work.

[github-actions]

Looks like this PR hasn't been updated in a while so we're going to go ahead and mark this as Stale.
Feel free to remove the Stale label if you feel this was a mistake.
If you are unable to remove the Stale label please contact a maintainer in order to do so.
If you want the bot to never mark this PR stale again, add the no-stale label.
Stale pull requests will automatically be closed after 30 days of inactivity.