Create build_wheels_aarch64_linux.yml

[msaroufim]

See #1082 and #1076

AO seems to mostly work fine with torch.compile support for ARM so might as well ship the binaries

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/1083

• 📄 Preview Python docs built from this PR

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

✅ No Failures

As of commit 874db1c with merge base 48bc81c ():
💚 Looks good so far! There are no failures yet. 💚

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

[msaroufim]

@atalman do you know why the AARCH jobs here are running a dry-run? Do I need to merge this code to main to do binary validation? I checked the torchvision jobs FWIW and they don't seem to postprocess aarch64 binaries with anything special. So if this looks good mind stamping and I can merge and test?

The wheel looks fine

+ aws s3 cp --dryrun dist/torchao-0.7.0.dev20241015-cp39-cp39-linux_aarch64.whl s3://pytorch/whl/nightly/cu124/ --acl public-read
(dryrun) upload: dist/torchao-0.7.0.dev20241015-cp39-cp39-linux_aarch64.whl to s3://pytorch/whl/nightly/cu124/torchao-0.7.0.dev20241015-cp39-cp39-linux_aarch64.whl

[atalman]

lgtm

[msaroufim]

Confirming this works when I tested it on my raspberry pi 5

ERROR: ResolutionImpossible: for help visit https://pip.pypa.io/en/latest/topics/dependency-resolution/#dealing-with-dependency-conflicts
(myvenv) marksaroufim@rpi5:~/Dev $ pip install --pre torchao --index-url https://download.pytorch.org/whl/nightly/cpu --force-reinstall
Looking in indexes: https://download.pytorch.org/whl/nightly/cpu, https://www.piwheels.org/simple
Collecting torchao
  Downloading https://download.pytorch.org/whl/nightly/cpu/torchao-0.7.0.dev20241017-cp311-cp311-linux_aarch64.whl (515 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 515.1/515.1 kB 584.4 kB/s eta 0:00:00
Installing collected packages: torchao
  Attempting uninstall: torchao
    Found existing installation: torchao 0.0.3
    Uninstalling torchao-0.0.3:
      Successfully uninstalled torchao-0.0.3
Successfully installed torchao-0.7.0.dev20241017
(myvenv) marksaroufim@rpi5:~/Dev $ python
Python 3.11.2 (main, Aug 26 2024, 07:20:54) [GCC 12.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torchao
>>> $