Update aoti APIs

[angelayi]

No description provided.

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/torchchat/1455

• 📄 Preview Python docs built from this PR

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

❌ 1 New Failure, 2 Unrelated Failures

As of commit 195aba8 with merge base 07768cf ():

NEW FAILURE - The following job has failed:

• pull / test-build-runner-et-android / linux-job (gh)
  RuntimeError: Command docker exec -t f02db0688ec6e6ea73a6b530a515f4cb603a3b58793517a090e077925dd3a408 /exec failed with exit code 1

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

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

• pull / test-tinystories-executorch (16-core-ubuntu) (gh) (trunk failure)
  ##[error]The operation was canceled.
• pull / test-tinystories-executorch (macos-14-xlarge) (gh) (trunk failure)
  RuntimeError: Failed to load ET compiled /Users/runner/work/torchchat/torchchat/stories15M.pte

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

[Jack-Khuu]

Thanks for the update, mind adding a bit of context to the description?

Looks like the error is legit? An arg mismatch

Edit: some of the ET ones are not related and are being tackled in #1459