Skip to content

Add CI for Arm Linux #2211

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 3 commits into from
May 15, 2025
Merged

Add CI for Arm Linux #2211

merged 3 commits into from
May 15, 2025

Conversation

metascroy
Copy link
Contributor

@metascroy metascroy commented May 14, 2025

@vctrmn added support for Arm linux in #2169 tested locally. Here we add CI for Arm linux to complement the Arm MacOS CI.

Copy link

pytorch-bot bot commented May 14, 2025

🔗 Helpful Links

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

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

❗ 1 Active SEVs

There are 1 currently active SEVs. If your PR is affected, please view them below:

✅ No Failures

As of commit dacbb2a with merge base 554cb60 (image):
💚 Looks good so far! There are no failures yet. 💚

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

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label May 14, 2025
@metascroy metascroy added ci topic: not user facing Use this tag if you don't want this PR to show up in release notes labels May 15, 2025
@metascroy metascroy marked this pull request as ready for review May 15, 2025 00:21
@metascroy metascroy changed the title Update torchao_experimental_test.yml Add CI for Arm Linux May 15, 2025
@metascroy metascroy requested a review from digantdesai May 15, 2025 00:23
@vctrmn
Copy link
Contributor

vctrmn commented May 15, 2025

@metascroy Great!

For my information, what kind of instances are used for linux.arm64.2xlarge? I am just curious about the CPU architecture (available flags from lscpu).

Would there be any way in the future to also test with TORCHAO_ENABLE_ARM_I8MM=1?

What do you think of github ARM runners that are in preview? (https://github.blog/news-insights/product-news/arm64-on-github-actions-powering-faster-more-efficient-build-systems/)

Just being curious

@metascroy
Copy link
Contributor Author

@metascroy Great!

For my information, what kind of instances are used for linux.arm64.2xlarge? I am just curious about the CPU architecture (available flags from lscpu).

Would there be any way in the future to also test with TORCHAO_ENABLE_ARM_I8MM=1?

What do you think of github ARM runners that are in preview? (https://github.blog/news-insights/product-news/arm64-on-github-actions-powering-faster-more-efficient-build-systems/)

Just being curious

This is an AWS EC2 graviton2 instance (Amazon EC2 T4g), which I do not think supports I8MM. In future, we could try adding a graviton3 instance for i8mm.

@msaroufim msaroufim self-requested a review May 15, 2025 20:37
@metascroy metascroy merged commit f04ff57 into main May 15, 2025
20 of 21 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
ci CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: not user facing Use this tag if you don't want this PR to show up in release notes
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants