Skip to content

Use upstream DTensor scatter strategy#502

Closed
aditvenk wants to merge 1 commit into
aditvenk/stack/13from
aditvenk/stack/14
Closed

Use upstream DTensor scatter strategy#502
aditvenk wants to merge 1 commit into
aditvenk/stack/13from
aditvenk/stack/14

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 scatter.src strategy that preserves AutoParallel's local sharding cases and also validates src shape compatibility, so the local scatter rule can be removed.

Authored with Claude.

stack-info: PR: #502, branch: aditvenk/stack/14
@aditvenk aditvenk marked this pull request as draft June 15, 2026 16:27
@aditvenk aditvenk changed the base branch from aditvenk/stack/13 to main June 15, 2026 16:27
@aditvenk aditvenk force-pushed the aditvenk/stack/14 branch from 2501332 to b90e952 Compare June 15, 2026 16:27
@aditvenk aditvenk changed the base branch from main to aditvenk/stack/13 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