Skip to content

[DTensor] make expand_to_full_mesh_op_strategy filter incompatible out= strategies#172420

Closed
wconstab wants to merge 8 commits intogh/wconstab/497/basefrom
gh/wconstab/497/head
Closed

[DTensor] make expand_to_full_mesh_op_strategy filter incompatible out= strategies#172420
wconstab wants to merge 8 commits intogh/wconstab/497/basefrom
gh/wconstab/497/head

Conversation

@pytorch-bot
Copy link
Copy Markdown

pytorch-bot Bot commented Jan 14, 2026

🔗 Helpful Links

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

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

❌ 7 New Failures, 1 Pending, 1 Unrelated Failure

As of commit 2647e22 with merge base 879fd46 (image):

NEW FAILURES - The following jobs have failed:

FLAKY - The following job failed but was likely due to flakiness present on trunk:

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

if (
op_schema.is_out_variant_op()
and "out" in op_schema.kwargs_schema
and isinstance(op_schema.kwargs_schema["out"], OpStrategy)
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: is there a TupleStrategy case we need to consider?

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.

are you talking about a case where out= provides a list of output tensors instead of a single tensor? I am not sure if such an op exists, trying to find one

@wconstab
Copy link
Copy Markdown
Contributor Author

@pytorchbot merge

@pytorch-bot pytorch-bot Bot added the ciflow/trunk Trigger trunk jobs on your pull request label Jan 16, 2026
@pytorchmergebot
Copy link
Copy Markdown
Collaborator

Merge failed

Reason: PR 172477 is out of sync with the corresponding revision 27ce290 on branch gh/wconstab/499/orig that would be merged into main. This usually happens because there is a non ghstack change in the PR. Please sync them and try again (ex. make the changes on origin/gh/wconstab/497/orig and run ghstack).

Details for Dev Infra team Raised by workflow job

SergeyTyshkevich pushed a commit to SergeyTyshkevich/chart2 that referenced this pull request Jan 19, 2026
…mpatible out= strategies"

[ghstack-poisoned]
@pytorchmergebot
Copy link
Copy Markdown
Collaborator

Starting merge as part of PR stack under #172477

…mpatible out= strategies"

[ghstack-poisoned]
@pytorchmergebot
Copy link
Copy Markdown
Collaborator

Starting merge as part of PR stack under #172477

1 similar comment
@pytorchmergebot
Copy link
Copy Markdown
Collaborator

Starting merge as part of PR stack under #172477

pytorchmergebot pushed a commit that referenced this pull request Jan 20, 2026
This enables the inplace filtering logic that skips strategies with
incompatible input placements.

Previously, inplace ops were able to generate expanded strategies
incompatible with the current input placements, which aren't allowed to
be modified by redistribution.
Pull Request resolved: #172477
Approved by: https://github.com/zpcore
ghstack dependencies: #172420
suncapitalllc007-star pushed a commit to suncapitalllc007-star/pytorch that referenced this pull request Jan 25, 2026
suncapitalllc007-star pushed a commit to suncapitalllc007-star/pytorch that referenced this pull request Jan 25, 2026
@github-actions github-actions Bot deleted the gh/wconstab/497/head branch February 20, 2026 02:22
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/inductor ciflow/trunk Trigger trunk jobs on your pull request Merged release notes: distributed (dtensor) release notes category

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants