Skip to content

add batching rule for torch.Tensor.scatter_add_#150543

Closed
guilhermeleobas wants to merge 4 commits intogh/guilhermeleobas/117/basefrom
gh/guilhermeleobas/117/head
Closed

add batching rule for torch.Tensor.scatter_add_#150543
guilhermeleobas wants to merge 4 commits intogh/guilhermeleobas/117/basefrom
gh/guilhermeleobas/117/head

Conversation

@guilhermeleobas
Copy link
Copy Markdown
Collaborator

@guilhermeleobas guilhermeleobas commented Apr 2, 2025

[ghstack-poisoned]
@pytorch-bot
Copy link
Copy Markdown

pytorch-bot bot commented Apr 2, 2025

🔗 Helpful Links

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

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

✅ No Failures

As of commit 3311987 with merge base 3da14d3 (image):
💚 Looks good so far! There are no failures yet. 💚

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

guilhermeleobas added a commit that referenced this pull request Apr 2, 2025
ghstack-source-id: e45acb8
Pull Request resolved: #150543
@guilhermeleobas guilhermeleobas added the module: functorch Pertaining to torch.func or pytorch/functorch label Apr 2, 2025
@guilhermeleobas guilhermeleobas linked an issue Apr 2, 2025 that may be closed by this pull request
@guilhermeleobas guilhermeleobas added the release notes: torch.func release notes category for torch.vmap or torch.func.* APIs label Apr 2, 2025
[ghstack-poisoned]
guilhermeleobas added a commit that referenced this pull request Apr 2, 2025
ghstack-source-id: c665cc4
Pull Request resolved: #150543
@guilhermeleobas guilhermeleobas self-assigned this Apr 2, 2025
@guilhermeleobas guilhermeleobas marked this pull request as ready for review April 2, 2025 19:45
int64_t dim,
const Tensor& index, std::optional<int64_t> index_bdim,
const Tensor& src, std::optional<int64_t> src_bdim) {
auto self_ = self.clone(at::MemoryFormat::Preserve);
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.

why do we need the clone now?

Copy link
Copy Markdown
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

We don't.

[ghstack-poisoned]
guilhermeleobas added a commit that referenced this pull request Apr 7, 2025
ghstack-source-id: 8ef4573
Pull Request resolved: #150543
@guilhermeleobas
Copy link
Copy Markdown
Collaborator Author

@pytorchbot merge -r

@pytorch-bot pytorch-bot bot added the ciflow/trunk Trigger trunk jobs on your pull request label Apr 8, 2025
@pytorchmergebot
Copy link
Copy Markdown
Collaborator

@pytorchbot started a rebase job onto refs/remotes/origin/viable/strict. Check the current status here

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

Successfully rebased gh/guilhermeleobas/117/orig onto refs/remotes/origin/viable/strict, please pull locally before adding more changes (for example, via ghstack checkout https://github.com/pytorch/pytorch/pull/150543)

pytorchmergebot pushed a commit that referenced this pull request Apr 8, 2025
ghstack-source-id: 0936707
Pull Request resolved: #150543
@pytorchmergebot
Copy link
Copy Markdown
Collaborator

Merge started

Your change will be merged once all checks pass (ETA 0-4 Hours).

Learn more about merging in the wiki.

Questions? Feedback? Please reach out to the PyTorch DevX Team

Advanced Debugging
Check the merge workflow status
here

timocafe pushed a commit to timocafe/pytorch that referenced this pull request Apr 16, 2025
amathewc pushed a commit to amathewc/pytorch that referenced this pull request Apr 17, 2025
@github-actions github-actions bot deleted the gh/guilhermeleobas/117/head branch May 16, 2025 02:17
frgossen added a commit that referenced this pull request Mar 11, 2026
The underlying bug (eager vs AOTDispatcher output mismatch for
as_strided_scatter) was fixed by #150543. Remove the stale skip.

Closes #85879

[ghstack-poisoned]
frgossen added a commit that referenced this pull request Mar 11, 2026
The underlying bug (eager vs AOTDispatcher output mismatch for
as_strided_scatter) was fixed by #150543. Remove the stale skip.

Closes #85879

ghstack-source-id: 88311bf
Pull Request resolved: #177203
pytorchmergebot pushed a commit that referenced this pull request Mar 12, 2026
The underlying bug (eager vs AOTDispatcher output mismatch for
as_strided_scatter) was fixed by #150543. Remove the stale skip.

Closes #85879
Pull Request resolved: #177203
Approved by: https://github.com/aorenste, https://github.com/zou3519
EmanueleCoradin pushed a commit to EmanueleCoradin/pytorch that referenced this pull request Mar 30, 2026
…77203)

The underlying bug (eager vs AOTDispatcher output mismatch for
as_strided_scatter) was fixed by pytorch#150543. Remove the stale skip.

Closes pytorch#85879
Pull Request resolved: pytorch#177203
Approved by: https://github.com/aorenste, https://github.com/zou3519
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/trunk Trigger trunk jobs on your pull request Merged module: functorch Pertaining to torch.func or pytorch/functorch open source release notes: torch.func release notes category for torch.vmap or torch.func.* APIs

Projects

None yet

Development

Successfully merging this pull request may close these issues.

batching rule for aten::scatter_add_

4 participants