Example external model to be used by the ahead of time arm compiler

[Juanfi8]

Example of an external model to be loaded as a module by the aot_arm_compiler.py.
Useful for documentation.

@zingo

cc @digantdesai @freddan80 @per @zingo @oscarandersson8218

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 8 New Failures, 4 Unrelated Failures

As of commit 3d100d7 with merge base a21022c ():

NEW FAILURES - The following jobs have failed:

• pull / android / build-llm-demo / linux-job (gh)
  RuntimeError: Command docker exec -t 585ed7ca40ffd5fb4635d3b63e70cb073675b81d4be113b8b1b6d6964fc6646d /exec failed with exit code 1
• pull / test-eval_llama-mmlu-linux / linux-job (gh)
  RuntimeError: Command docker exec -t 1345ce2ef8c53f5f9be54a1d71ac38839d5200c4e5cfd5e846740d98b943b4a0 /exec failed with exit code 1
• pull / test-eval_llama-wikitext-linux / linux-job (gh)
  RuntimeError: Command docker exec -t f528707ad3c227e77313e7275a4c2fb018536196f2d1a07c4fe3b5539961035e /exec failed with exit code 1
• pull / test-llama_runner_eager-linux / linux-job (gh)
  RuntimeError: Command docker exec -t efbd0b60b92b73672e8f3ae4cb7fceeb215eb11e7b9c3535d82a7374e1952f4b /exec failed with exit code 1
• pull / test-llama-runner-linux (bf16, custom, linux.2xlarge, executorch-ubuntu-22.04-clang12) / linux-job (gh)
  RuntimeError: Command docker exec -t 9302a8a36629ef6417ad2b6280f9ea42bf4997cae7138b1a8fa54cbe2851f8b3 /exec failed with exit code 1
• pull / test-llama-runner-qnn-linux (fp32, qnn_16a16w, qnn) / linux-job (gh)
  RuntimeError: Command docker exec -t ec8a66553c152ce960980da96e6ccc10e80a60dd1acf640444cf562067eb7cb8 /exec failed with exit code 1
• pull / test-llama-runner-qnn-linux (fp32, qnn_8a8w, qnn) / linux-job (gh)
  RuntimeError: Command docker exec -t 5178a280ac0d6d47b0705950b5a564b75fe820f22eba2f7a4051dad4715b58a2 /exec failed with exit code 1
• pull / test-models-linux (phi_4_mini, portable, linux.4xlarge.memory) / linux-job (gh)
  RuntimeError: Command docker exec -t c2181f1f81ce46ed895805a5c911f5753f299ff563dac56264d656e3cbd7f60a /exec failed with exit code 1

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

• pull / unittest-editable / linux / linux-job (gh) (similar failure)
  examples/models/llama/tests/test_export_llama_lib.py::ExportLlamaLibTest::test_has_expected_ops_and_op_counts

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

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

• pull / unittest / linux / linux-job (gh) (trunk failure)
  examples/models/llama/tests/test_export_llama_lib.py::ExportLlamaLibTest::test_has_expected_ops_and_op_counts
• pull / unittest / macos / macos-job (gh) (trunk failure)
  examples/models/llama/tests/test_export_llama_lib.py::ExportLlamaLibTest::test_has_expected_ops_and_op_counts
• pull / unittest-editable / macos / macos-job (gh) (trunk failure)
  examples/models/llama/tests/test_export_llama_lib.py::ExportLlamaLibTest::test_has_expected_ops_and_op_counts

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

[Juanfi8]

@pytorchbot label "partner: arm"

[pytorch-bot]

To add the ciflow label ciflow/trunk please first approve the workflows that are awaiting approval (scroll to the bottom of this page).

This helps ensure we don't trigger CI on this PR until it is actually authorized to do so. Please ping one of the reviewers if you do not have access to approve and run workflows.

[zingo]

Nice patch looks great! Thanks. I just have a small request if it's OK?

As its only Arm that support this thing right now would it be OK to move the files to examples/arm/

e.g. from

examples/models/example_export_from_module

to

examples/arm/examples
or
examples/arm/example_modules
or keep the longer folder name if you like.

And as we probably will add a few more test models down the line maybe rename it from model.py to model_add.py (or add.py ) ?

Thanks for the help!

[zingo]

Thanks for the update!

[zingo]

All the llama test errors are unrelated and seen in other PRs right now.

[zingo]

@Juanfi8 Almost ready to merge :)
There is just a smal lintrunner error see:
https://github.com/pytorch/executorch/actions/runs/15019836027/job/42206414796?pr=10810

lintrunner --force-color --all-files --tee-json=lint.json

Lint for examples/arm/example_modules/add.py:

Warning (UFMT) format
Run lintrunner -a to apply this patch.

You can run `lintrunner -a` to apply this patch.

 1   1 | import torch
 2   2 | 
     3 |+
 3   4 | class myModelAdd(torch.nn.Module):
 4   5 |     def __init__(self):
 5   6 |         super().__init__()

---

 7   8 |     def forward(self, x):
 8   9 |         return x + x
 9  10 | 
 9     |-ModelUnderTest=myModelAdd()
    11 |+
    12 |+ModelUnderTest = myModelAdd()
11  13 | ModelInputs = (torch.ones(5),)

[Juanfi8]

I hope this time is the one :)
Happy to contribute to the project.

[zingo]

Great work!