Arm backend: Fix amax test merge issue

[AdrianLundell]

cc @freddan80 @per @zingo @oscarandersson8218 @digantdesai

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 1 Awaiting Approval, 5 New Failures, 4 Pending

As of commit 1686153 with merge base c229dbc ():

AWAITING APPROVAL - The following workflow needs approval before CI can run:

• periodic (gh)

NEW FAILURES - The following jobs have failed:

• pull / test-samsung-models-linux / linux-job (gh)
  RuntimeError: Command docker exec -t 0499e726cac906b0c50e4dc4eaf364bcc0e17ab564cb81ccd52a292bf9bb1c1c /exec failed with exit code 1
• pull / test-samsung-quantmodels-linux / linux-job (gh)
  RuntimeError: Command docker exec -t 888f5f0b4eee31ef9b03d0b5bcf21c783b4e44617f531b12ad2c3054b60f749a /exec failed with exit code 1
• trunk / test-huggingface-transformers-macos (qwen3-0.6b|coreml_fp32_gpu|--quantize) / macos-job (gh)
  The process '/opt/homebrew/bin/git' failed with exit code 128
• trunk / test-mcu-cortex-m-backend / linux-job (gh)
  RuntimeError: Command docker exec -t cb6d835ed921c4e05a2faccf86afa601755551c11b25c3fc3bfb41707c13e0de /exec failed with exit code 1
• trunk / test-models-macos-coreml (mobilebert) / macos-job (gh)
  RuntimeError: Command bash /Users/ec2-user/runner/_work/_temp/exec_script failed with exit code 1

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

[AdrianLundell]

Fails unrelated