Skip to content

torch.ops.fbgemm.scatter_add_along_first_dim.. #3720

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 1 commit into from

Conversation

levendlee
Copy link
Member

Summary:
TMA based scatter_add operation optimized for large shapes.
Hyperparameters could be finetuned for better performance. However, the expected headroom is small.

Differential Revision: D69957147

Copy link

netlify bot commented Feb 21, 2025

Deploy Preview for pytorch-fbgemm-docs ready!

Name Link
🔨 Latest commit 2a35571
🔍 Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/67ba3ba16ad9cc00088e26d5
😎 Deploy Preview https://deploy-preview-3720--pytorch-fbgemm-docs.netlify.app
📱 Preview on mobile
Toggle QR Code...

QR Code

Use your smartphone camera to open QR code link.

To edit notification comments on pull requests, go to your Netlify site configuration.

@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D69957147

@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D69957147

levendlee added a commit to levendlee/FBGEMM that referenced this pull request Feb 21, 2025
Summary:
Pull Request resolved: pytorch#3720

X-link: facebookresearch/FBGEMM#804

TMA based scatter_add operation optimized for large shapes.
Hyperparameters could be finetuned for better performance. However, the expected headroom is small.

Differential Revision: D69957147
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D69957147

levendlee added a commit to levendlee/FBGEMM that referenced this pull request Feb 21, 2025
Summary:
Pull Request resolved: pytorch#3720

X-link: facebookresearch/FBGEMM#804

TMA based scatter_add operation optimized for large shapes.
Hyperparameters could be finetuned for better performance. However, the expected headroom is small.

Differential Revision: D69957147
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D69957147

levendlee added a commit to levendlee/FBGEMM that referenced this pull request Feb 21, 2025
Summary:
Pull Request resolved: pytorch#3720

X-link: facebookresearch/FBGEMM#804

TMA based scatter_add operation optimized for large shapes.
Hyperparameters could be finetuned for better performance. However, the expected headroom is small.

Differential Revision: D69957147
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D69957147

levendlee added a commit to levendlee/FBGEMM that referenced this pull request Feb 21, 2025
Summary:
Pull Request resolved: pytorch#3720

X-link: facebookresearch/FBGEMM#804

TMA based scatter_add operation optimized for large shapes.
Hyperparameters could be finetuned for better performance. However, the expected headroom is small.

Differential Revision: D69957147
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D69957147

levendlee added a commit to levendlee/FBGEMM that referenced this pull request Feb 21, 2025
Summary:
Pull Request resolved: pytorch#3720

X-link: facebookresearch/FBGEMM#804

TMA based scatter_add operation optimized for large shapes.
Hyperparameters could be finetuned for better performance. However, the expected headroom is small.

Differential Revision: D69957147
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D69957147

levendlee added a commit to levendlee/FBGEMM that referenced this pull request Feb 21, 2025
Summary:
Pull Request resolved: pytorch#3720

X-link: facebookresearch/FBGEMM#804

TMA based scatter_add operation optimized for large shapes.
Hyperparameters could be finetuned for better performance. However, the expected headroom is small.

Differential Revision: D69957147
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D69957147

levendlee added a commit to levendlee/FBGEMM that referenced this pull request Feb 21, 2025
Summary:
Pull Request resolved: pytorch#3720

X-link: facebookresearch/FBGEMM#804

TMA based scatter_add operation optimized for large shapes.
Hyperparameters could be finetuned for better performance. However, the expected headroom is small.

Differential Revision: D69957147
Summary:
Pull Request resolved: pytorch#3720

X-link: facebookresearch/FBGEMM#804

TMA based scatter_add operation optimized for large shapes.
Hyperparameters could be finetuned for better performance. However, the expected headroom is small.

Reviewed By: jianyuh

Differential Revision: D69957147
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D69957147

@facebook-github-bot
Copy link
Contributor

This pull request has been merged in 221c2aa.

q10 pushed a commit to q10/FBGEMM that referenced this pull request Apr 10, 2025
Summary:
X-link: pytorch#3720

Pull Request resolved: facebookresearch/FBGEMM#804

TMA based scatter_add operation optimized for large shapes.
Hyperparameters could be finetuned for better performance. However, the expected headroom is small.

Reviewed By: jianyuh

Differential Revision: D69957147

fbshipit-source-id: 17aefbeb379976b651e7abd123b885d48d479d1a
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants