Skip to content

Use upstream DTensor select scatter strategy#497

Closed
aditvenk wants to merge 1 commit into
aditvenk/stack/8from
aditvenk/stack/9
Closed

Use upstream DTensor select scatter strategy#497
aditvenk wants to merge 1 commit into
aditvenk/stack/8from
aditvenk/stack/9

Conversation

@fmassa fmassa left a comment

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.

LGTM, thanks!

PyTorch now provides a single-dim strategy for aten.select_scatter.default with the same non-selected-dimension sharding mapping, so AutoParallel can remove its local rule.

Authored with Claude.

stack-info: PR: #497, branch: aditvenk/stack/9
@aditvenk aditvenk marked this pull request as draft June 15, 2026 16:26
@aditvenk aditvenk changed the base branch from aditvenk/stack/8 to main June 15, 2026 16:26
@aditvenk aditvenk changed the base branch from main to aditvenk/stack/8 June 15, 2026 16:27
@aditvenk aditvenk marked this pull request as ready for review June 15, 2026 16:28
@aditvenk

Copy link
Copy Markdown
Contributor Author

closing in favor of #505

@aditvenk aditvenk closed this Jun 15, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Meta Open Source bot.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants