Arm backend: Fix bug in memory allocation

[gggekov]

If running multiple inferences on the same Ethos-U custom delegate, we need to reset the temp_allocator after every inference. Otherwise, we keep allocating more memory within the temp_allocator rather than reuse the temp_allocator between inferences.

cc @digantdesai @freddan80 @per @zingo @oscarandersson8218

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 11 New Failures, 7 Cancelled Jobs, 3 Unrelated Failures

As of commit 8abb465 with merge base 4d6209b ():

NEW FAILURES - The following jobs have failed:

• Build Windows Wheels / pytorch/executorch / build-wheel-py3_10-cpu (gh)
  RuntimeError: Failed to install QNN SDK. Please check the logs above.
• Build Windows Wheels / pytorch/executorch / upload / upload-wheel-py3_10-cpu (gh)
  Unable to download artifact(s): Artifact not found for name: pytorch_executorch__3.10_cpu_x64
• pull / test-moshi-linux / linux-job (gh)
  RuntimeError: Command docker exec -t c195c6612e1ad6ec6167c573148b1bb611ae26c8cbc6d6f5100f2c8e87b77231 /exec failed with exit code 1
• pull / test-qnn-wheel-packages-linux (3.10) / linux-job (gh)
  RuntimeError: Command docker exec -t f5314166167cf51a1bd2cc27ac635854984b0e3fe30b3a816e66ed57ba93aed4 /exec failed with exit code 127
• pull / test-qnn-wheel-packages-linux (3.11) / linux-job (gh)
  RuntimeError: Command docker exec -t 5608d9bc0d93f3781fb7ad06024bb43f932902c7b56cbc490c6f3315f34db11a /exec failed with exit code 127
• pull / test-qnn-wheel-packages-linux (3.12) / linux-job (gh)
  RuntimeError: Command docker exec -t 74c656850400f8311feba35d0e6a96c2c99e82afb961b838184dccdd34faf00d /exec failed with exit code 127
• pull / test-samsung-models-linux / linux-job (gh)
  RuntimeError: Command docker exec -t 052775be9349affe51d1ba22cf41b3d10c3467c322aede83c84ae6f1c7ff8136 /exec failed with exit code 23
• trunk / test-arm-backend (test_pytest_ops_ethosu_fvp) / linux-job (gh)
  RuntimeError: Command docker exec -t 7047347df69ed98801c5ec42dad52952a24d7f107fb88a0bd3aa052a4951468f /exec failed with exit code 1
• trunk / test-models-macos-coreml (ic3) / macos-job (gh)
  RuntimeError: Command bash /Users/ec2-user/runner/_work/_temp/exec_script failed with exit code 139
• trunk / unittest-release / linux / linux-job (gh)
  extension/llm/tokenizers/test/test_hf_tokenizer.py::TestHfTokenizer::test_llama3_2_1b
• trunk / unittest-release / macos / macos-job (gh)
  backends/xnnpack/test/ops/test_conv2d.py::TestConv2d::test_fp16_conv2d

CANCELLED JOBS - The following jobs were cancelled. Please retry:

• periodic / test-models-linux (buck2, mv3, portable, linux.2xlarge, 90) / linux-job (gh)
  ##[error]The operation was canceled.
• periodic / test-models-linux (buck2, mv3, xnnpack-quantization-delegation, linux.2xlarge, 90) / linux-job (gh)
  ##[error]The operation was canceled.
• periodic / test-models-linux (cmake, mv3, portable, linux.2xlarge, 90) / linux-job (gh)
  ##[error]The operation was canceled.
• periodic / test-models-linux (cmake, mv3, xnnpack-quantization-delegation, linux.2xlarge, 90) / linux-job (gh)
  ##[error]The operation was canceled.
• periodic / test-models-linux (cmake, vit, portable, linux.2xlarge, 90) / linux-job (gh)
  ##[error]The operation was canceled.
• periodic / test-models-linux (cmake, vit, xnnpack-quantization-delegation, linux.2xlarge, 90) / linux-job (gh)
  ##[error]The operation was canceled.
• pull / test-openvino-linux / linux-job (gh)

FLAKY - The following job failed but was likely due to flakiness present on trunk:

• pull / test-setup-linux-gcc / linux-job (gh) (similar failure)
  ##[error]The operation was canceled.

BROKEN TRUNK - The following jobs failed but was present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

• pull / test-binary-size-linux-gcc / linux-job (gh) (trunk failure)
  ##[error]The operation was canceled.
• trunk / test-models-macos-cpu (mv3, xnnpack-quantization-delegation) / macos-job (gh) (trunk failure)
  RuntimeError: Command bash /Users/ec2-user/runner/_work/_temp/exec_script failed with exit code 139

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

[facebook-github-bot]

@mergennachin has imported this pull request. If you are a Meta employee, you can view this in D82220111.