Skip to content

Use upstream DTensor convolution backward strategy#501

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

Use upstream DTensor convolution backward strategy#501
aditvenk wants to merge 1 commit into
aditvenk/stack/12from
aditvenk/stack/13

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.convolution_backward.default with AutoParallel's batch-shard and partial-gradient mapping, so the local backward convolution rule can be removed.

Authored with Claude.

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