Skip to content

Add --no-patches flag option to get-source.sh + freeze pytest<9 to address failing PyTorch unit tests#456

Open
puneetmatharu wants to merge 1 commit into
ARM-software:mainfrom
puneetmatharu:add-no-patches-flag-to-get-source-sh
Open

Add --no-patches flag option to get-source.sh + freeze pytest<9 to address failing PyTorch unit tests#456
puneetmatharu wants to merge 1 commit into
ARM-software:mainfrom
puneetmatharu:add-no-patches-flag-to-get-source-sh

Conversation

@puneetmatharu
Copy link
Copy Markdown
Contributor

No description provided.

Copilot AI review requested due to automatic review settings May 11, 2026 07:28
Copy link
Copy Markdown

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

Adds an opt-out switch to skip applying WIP GitHub patches during source checkout for the AArch64 TensorFlow and PyTorch build scripts, and documents the new option in each component’s changelog.

Changes:

  • Gate TensorFlow patch application in tensorflow-aarch64/get-source.sh behind a --no-patches flag.
  • Gate PyTorch and nested oneDNN patch application in pytorch-aarch64/get-source.sh behind the same --no-patches flag.
  • Update both framework changelogs to note the new --no-patches option.

Reviewed changes

Copilot reviewed 4 out of 4 changed files in this pull request and generated 3 comments.

File Description
ML-Frameworks/tensorflow-aarch64/get-source.sh Adds --no-patches conditional around TensorFlow patch application.
ML-Frameworks/tensorflow-aarch64/CHANGELOG.md Documents the new TensorFlow --no-patches option.
ML-Frameworks/pytorch-aarch64/get-source.sh Adds --no-patches conditional around PyTorch and oneDNN patch application.
ML-Frameworks/pytorch-aarch64/CHANGELOG.md Documents the new PyTorch/oneDNN --no-patches option.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Comment thread ML-Frameworks/tensorflow-aarch64/get-source.sh
Comment thread ML-Frameworks/pytorch-aarch64/get-source.sh Outdated
Comment thread ML-Frameworks/pytorch-aarch64/get-source.sh Outdated
@puneetmatharu puneetmatharu force-pushed the add-no-patches-flag-to-get-source-sh branch 2 times, most recently from 19df3bb to c7c9a87 Compare May 11, 2026 14:09
@puneetmatharu puneetmatharu changed the title Add --no-patches flag option to get-source.sh Add --no-patches flag option to get-source.sh + freeze pytest<9 to address failing PyTorch unit tests May 12, 2026
@puneetmatharu puneetmatharu requested review from jondea and nSircombe May 13, 2026 08:58
@puneetmatharu
Copy link
Copy Markdown
Contributor Author

Probably a good idea to try and get this PR very soon to fix the failing CI.

@jondea
Copy link
Copy Markdown
Contributor

jondea commented May 13, 2026

This is definitely my fault for not fully thinking out this feature request, but I think I envisioned

  • --no-patches also not modifying the oneDNN version, i.e. just building an upstream PyTorch. I'm not sure whether anyone would be interested in the inbetween (modifying oneDNN version but no patches)
  • Able to specify --no-patches from build.sh. I think this is the primary endpoint, I think it would be nice to be accessible from there

@puneetmatharu puneetmatharu force-pushed the add-no-patches-flag-to-get-source-sh branch 3 times, most recently from cb4a8bf to 8c39ce7 Compare May 13, 2026 14:39
@puneetmatharu
Copy link
Copy Markdown
Contributor Author

Done!

Copy link
Copy Markdown
Contributor

@nSircombe nSircombe left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

hummm. I think my 'assumption' about what --no-patches is differs from @jondea.

I assumed it would just skip he patches. Keeping hashes the same.

I think using the upstream hashes would / should be a separate flag, so getting the behaviour @jondea is shooting for would need --no-patches --original-hashes or similar.

### Added
- TorchAO installation via nightly AArch64 wheel in Dockerfile
- Adds a `--no-patches` option to `build.sh` and `get-source.sh` to disable patch application and use PyTorch's upstream submodule
> versions for ideep, oneDNN, and KleidiAI.
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

nit: previously for long lines we've just wrapped and indented to line up with the first character in the line above. See L89 and L90.

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Oh this is odd. I'm not sure how the > character got added... I'll fix it.

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Fixed!

@puneetmatharu puneetmatharu force-pushed the add-no-patches-flag-to-get-source-sh branch from 8c39ce7 to ddf7376 Compare May 14, 2026 07:39
@puneetmatharu puneetmatharu force-pushed the add-no-patches-flag-to-get-source-sh branch from ddf7376 to bbc91ec Compare May 14, 2026 11:37
@puneetmatharu
Copy link
Copy Markdown
Contributor Author

Okay @nSircombe @jondea I've opted for a --source-variant {upstream,pinned,patched} option to have the flexibility to use the three different variants we've discussed. Thoughts?

Copy link
Copy Markdown
Contributor

@nSircombe nSircombe left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The PyTorch changes look fine to me. We should use the same flags for TF too though.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants