Skip to content

Enable Partial Rowwise Adam in SSD TBE #4525

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

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

q10
Copy link
Contributor

@q10 q10 commented Jul 18, 2025

Summary: This diff enables Partial Rowwise Adam in SSD TBE, though just for the non KV-ZCH use case.

Reviewed By: emlin

Differential Revision: D77880362

Copy link

netlify bot commented Jul 18, 2025

Deploy Preview for pytorch-fbgemm-docs ready!

Name Link
🔨 Latest commit c27f02f
🔍 Latest deploy log https://app.netlify.com/projects/pytorch-fbgemm-docs/deploys/687e7f4f28e82e00084121de
😎 Deploy Preview https://deploy-preview-4525--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 project configuration.

@meta-cla meta-cla bot added the cla signed label Jul 18, 2025
@facebook-github-bot
Copy link
Contributor

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

@q10 q10 force-pushed the export-D77880362 branch 2 times, most recently from 8fa545c to 055127f Compare July 21, 2025 17:46
@facebook-github-bot
Copy link
Contributor

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

@q10 q10 force-pushed the export-D77880362 branch from 055127f to 618231f Compare July 21, 2025 17:49
Summary:
Pull Request resolved: pytorch#4525

X-link: facebookresearch/FBGEMM#1572

This diff enables Partial Rowwise Adam in SSD TBE, though just for the non KV-ZCH use case.

Reviewed By: emlin

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

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

@q10 q10 force-pushed the export-D77880362 branch from 618231f to c27f02f Compare July 21, 2025 17:56
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.

2 participants