Skip to content

Consolidate pre-grad pass timing tests into parametrized test#177731

Open
frgossen wants to merge 5 commits intogh/frgossen/15/basefrom
gh/frgossen/15/head
Open

Consolidate pre-grad pass timing tests into parametrized test#177731
frgossen wants to merge 5 commits intogh/frgossen/15/basefrom
gh/frgossen/15/head

Conversation

@frgossen
Copy link
Contributor

@frgossen frgossen commented Mar 18, 2026

Stack from ghstack (oldest at bottom):

Replace separate test_pre_grad_passes_late_timing,
test_pre_grad_passes_early_timing, test_pre_grad_passes_default_timing_with_uuid,
and test_pre_grad_passes_default_timing_without_uuid with a single parametrized
test_pre_grad_passes_timing that covers all timing/pass combinations.

Authored with Claude.

cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @kadeng @chauhang @amjames @Lucaskabela @jataylo

Replace separate test_pre_grad_passes_late_timing,
test_pre_grad_passes_early_timing, test_pre_grad_passes_default_timing_with_uuid,
and test_pre_grad_passes_default_timing_without_uuid with a single parametrized
test_pre_grad_passes_timing that covers all timing/pass combinations.

Authored with Claude.

[ghstack-poisoned]
@pytorch-bot
Copy link

pytorch-bot bot commented Mar 18, 2026

🔗 Helpful Links

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

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

❌ 2 New Failures, 2 Unrelated Failures

As of commit aee3673 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.

frgossen added a commit that referenced this pull request Mar 18, 2026
Replace separate test_pre_grad_passes_late_timing,
test_pre_grad_passes_early_timing, test_pre_grad_passes_default_timing_with_uuid,
and test_pre_grad_passes_default_timing_without_uuid with a single parametrized
test_pre_grad_passes_timing that covers all timing/pass combinations.

Authored with Claude.

ghstack-source-id: e0024d6
Pull Request resolved: #177731
…est"

Replace separate test_pre_grad_passes_late_timing,
test_pre_grad_passes_early_timing, test_pre_grad_passes_default_timing_with_uuid,
and test_pre_grad_passes_default_timing_without_uuid with a single parametrized
test_pre_grad_passes_timing that covers all timing/pass combinations.

Authored with Claude.

[ghstack-poisoned]
frgossen added a commit that referenced this pull request Mar 18, 2026
Replace separate test_pre_grad_passes_late_timing,
test_pre_grad_passes_early_timing, test_pre_grad_passes_default_timing_with_uuid,
and test_pre_grad_passes_default_timing_without_uuid with a single parametrized
test_pre_grad_passes_timing that covers all timing/pass combinations.

Authored with Claude.

ghstack-source-id: 4d8c280
Pull Request resolved: #177731
@frgossen frgossen added the module: aotdispatch umbrella label for AOTAutograd issues label Mar 18, 2026
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.

What does this look like from a test running perspective? I'm concerned about the ability to run one of these subtests locally in isolation - does pytest just use a simple string to identify each subtest (like "test_pre_grad_passes_timing_early_custom_pre_grad_pass_remove_ident_muls_wo_uuid_1_2")?

…est"

Replace separate test_pre_grad_passes_late_timing,
test_pre_grad_passes_early_timing, test_pre_grad_passes_default_timing_with_uuid,
and test_pre_grad_passes_default_timing_without_uuid with a single parametrized
test_pre_grad_passes_timing that covers all timing/pass combinations.

Authored with Claude.

cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx kadeng chauhang amjames Lucaskabela jataylo

[ghstack-poisoned]
…est"

Replace separate test_pre_grad_passes_late_timing,
test_pre_grad_passes_early_timing, test_pre_grad_passes_default_timing_with_uuid,
and test_pre_grad_passes_default_timing_without_uuid with a single parametrized
test_pre_grad_passes_timing that covers all timing/pass combinations.

Authored with Claude.

cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx kadeng chauhang amjames Lucaskabela jataylo

[ghstack-poisoned]
@frgossen
Copy link
Contributor Author

What does this look like from a test running perspective? I'm concerned about the ability to run one of these subtests locally in isolation - does pytest just use a simple string to identify each subtest (like "test_pre_grad_passes_timing_early_custom_pre_grad_pass_remove_ident_muls_wo_uuid_1_2")?

Yes, you can run them all individually, e.g:
python test/dynamo/test_aot_autograd_cache.py -k test_pre_grad_passes_timing_default_with_uuid

…est"

Replace separate test_pre_grad_passes_late_timing,
test_pre_grad_passes_early_timing, test_pre_grad_passes_default_timing_with_uuid,
and test_pre_grad_passes_default_timing_without_uuid with a single parametrized
test_pre_grad_passes_timing that covers all timing/pass combinations.

Authored with Claude.

cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx kadeng chauhang amjames Lucaskabela jataylo

[ghstack-poisoned]
@pytorchmergebot
Copy link
Collaborator

Starting merge as part of PR stack under #177733

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

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants