Skip to content

Default pre_grad_pass_timing to "late" in fbcode#177690

Open
frgossen wants to merge 3 commits intogh/frgossen/14/basefrom
gh/frgossen/14/head
Open

Default pre_grad_pass_timing to "late" in fbcode#177690
frgossen wants to merge 3 commits intogh/frgossen/14/basefrom
gh/frgossen/14/head

Conversation

@frgossen
Copy link
Contributor

@frgossen frgossen commented Mar 17, 2026

This ensures pre-grad passes run after cache lookups in Meta-internal
builds, avoiding unnecessary pass execution on cache hits.

Authored with Claude.

[ghstack-poisoned]
@pytorch-bot
Copy link

pytorch-bot bot commented Mar 17, 2026

🔗 Helpful Links

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

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

❌ 2 New Failures, 2 Unrelated Failures

As of commit 6fb7ee5 with merge base e3cc9e6 (image):

NEW FAILURES - The following jobs have failed:

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

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

UNSTABLE - The following job is marked as unstable, possibly due to flakiness on trunk:

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

@pytorch-bot
Copy link

pytorch-bot bot commented Mar 17, 2026

This PR needs a release notes: label

If your changes are user facing and intended to be a part of release notes, please use a label starting with release notes:.

If not, please add the topic: not user facing label.

To add a label, you can comment to pytorchbot, for example
@pytorchbot label "topic: not user facing"

For more information, see
https://github.com/pytorch/pytorch/wiki/PyTorch-AutoLabel-Bot#why-categorize-for-release-notes-and-how-does-it-work.

frgossen added a commit that referenced this pull request Mar 17, 2026
This ensures pre-grad passes run after cache lookups in Meta-internal
builds, avoiding unnecessary pass execution on cache hits.

Authored with Claude.

ghstack-source-id: 811004c
Pull Request resolved: #177690
@frgossen frgossen requested review from aorenste and zou3519 March 17, 2026 21:06
@frgossen frgossen added the module: aotdispatch umbrella label for AOTAutograd issues label Mar 17, 2026
This ensures pre-grad passes run after cache lookups in Meta-internal
builds, avoiding unnecessary pass execution on cache hits.

Authored with Claude.

cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx ipiszy kadeng muchulee8 amjames chauhang aakhundov coconutruben jataylo

[ghstack-poisoned]
Copy link
Contributor

@aorenste aorenste left a comment

Choose a reason for hiding this comment

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

You should either do this from the fbcode side or import into fbcode so we can make sure no internal tests fail before landing.

Comment on lines +665 to +667
pre_grad_pass_timing: Literal["early", "late", "default"] = (
"late" if is_fbcode() else "default"
)
Copy link
Contributor

Choose a reason for hiding this comment

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

pretty sure I saw this in another PR too :P

Copy link
Contributor

Choose a reason for hiding this comment

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

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Sorry about the confusion. Factored this out of the other PR again so that I can test it separately before merging.

This ensures pre-grad passes run after cache lookups in Meta-internal
builds, avoiding unnecessary pass execution on cache hits.

Authored with Claude.

cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx ipiszy kadeng muchulee8 amjames chauhang aakhundov coconutruben jataylo

[ghstack-poisoned]
frgossen added a commit that referenced this pull request Mar 19, 2026
This ensures pre-grad passes run after cache lookups in Meta-internal
builds, avoiding unnecessary pass execution on cache hits.

Authored with Claude.

ghstack-source-id: 8ebd7cc
Pull Request resolved: #177690
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/inductor ciflow/torchtitan Run TorchTitan integration tests module: aotdispatch umbrella label for AOTAutograd issues module: inductor topic: not user facing topic category

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants