Skip to content

Add support for int32_t indices in TBE training (2G/N) #3377

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 2 commits into from

Conversation

q10
Copy link
Contributor

@q10 q10 commented Nov 14, 2024

Summary: - Add index_t support to TBE training backward kernels

Differential Revision: D65960050

@facebook-github-bot
Copy link
Contributor

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

Copy link

netlify bot commented Nov 14, 2024

Deploy Preview for pytorch-fbgemm-docs ready!

Name Link
🔨 Latest commit 6137460
🔍 Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/6786b9873847a800081e6a9c
😎 Deploy Preview https://deploy-preview-3377--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.

@q10 q10 force-pushed the export-D65960050 branch from 5a012a1 to 569c1a3 Compare December 14, 2024 22:29
@facebook-github-bot
Copy link
Contributor

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

@q10 q10 force-pushed the export-D65960050 branch from 569c1a3 to 6c41ce6 Compare December 14, 2024 22:30
@facebook-github-bot
Copy link
Contributor

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

q10 added a commit to q10/FBGEMM that referenced this pull request Dec 14, 2024
Summary:

X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

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

Hi @q10!

Thank you for your pull request.

We require contributors to sign our Contributor License Agreement, and yours needs attention.

You currently have a record in our system, but the CLA is no longer valid, and will need to be resubmitted.

Process

In order for us to review and merge your suggested changes, please sign at https://code.facebook.com/cla. If you are contributing on behalf of someone else (eg your employer), the individual CLA may not be sufficient and your employer may need to sign the corporate CLA.

Once the CLA is signed, our tooling will perform checks and validations. Afterwards, the pull request will be tagged with CLA signed. The tagging process may take up to 1 hour after signing. Please give it that time before contacting us about it.

If you have received this in error or have any questions, please contact us at [email protected]. Thanks!

@q10 q10 force-pushed the export-D65960050 branch from 6c41ce6 to 00bb1db Compare January 2, 2025 20:37
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 2, 2025
Summary:

X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

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

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

q10 added a commit to q10/FBGEMM that referenced this pull request Jan 2, 2025
Summary:

X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 3, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 3, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 6, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Differential Revision: D65960050
@q10 q10 force-pushed the export-D65960050 branch from 00bb1db to 0eebd00 Compare January 6, 2025 21:42
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 6, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

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

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

@q10 q10 force-pushed the export-D65960050 branch from 0eebd00 to 14b3b97 Compare January 6, 2025 21:43
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 6, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

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

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

q10 added a commit to q10/FBGEMM that referenced this pull request Jan 6, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 6, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 6, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 7, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 7, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 9, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Differential Revision: D65960050
@q10 q10 force-pushed the export-D65960050 branch from 14b3b97 to 33bc751 Compare January 9, 2025 22:21
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 10, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Reviewed By: basilwong

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 11, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Reviewed By: basilwong

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 11, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Reviewed By: basilwong

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 11, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Reviewed By: basilwong

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 14, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Reviewed By: basilwong

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 14, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Reviewed By: basilwong

Differential Revision: D65960050
q10 added 2 commits January 14, 2025 11:22
Summary:
X-link: facebookresearch/FBGEMM#621

X-link: facebookresearch/FBGEMM#466


- Add `index_t` support to TBE training backward kernels

Reviewed By: basilwong

Differential Revision: D65933410
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Reviewed By: basilwong

Differential Revision: D65960050
@q10 q10 force-pushed the export-D65960050 branch from 7f5b31f to 6137460 Compare January 14, 2025 19:22
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 14, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Reviewed By: basilwong

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 14, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Reviewed By: basilwong

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

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

q10 added a commit to q10/FBGEMM that referenced this pull request Jan 14, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Reviewed By: basilwong

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 14, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Reviewed By: basilwong

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 14, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Reviewed By: basilwong

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 14, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Reviewed By: basilwong

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 14, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Reviewed By: basilwong

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 14, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Reviewed By: basilwong

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 15, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Reviewed By: basilwong

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 15, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Reviewed By: basilwong

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 15, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Reviewed By: basilwong

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 15, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Reviewed By: basilwong

Differential Revision: D65960050
q10 added a commit to q10/FBGEMM that referenced this pull request Jan 15, 2025
Summary:
X-link: facebookresearch/FBGEMM#622


X-link: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Reviewed By: basilwong

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

This pull request has been merged in 704cbc0.

q10 added a commit to q10/FBGEMM that referenced this pull request Apr 10, 2025
Summary:
Pull Request resolved: facebookresearch/FBGEMM#622

X-link: pytorch#3377

Pull Request resolved: facebookresearch/FBGEMM#468

- Add `index_t` support to TBE training backward kernels

Reviewed By: basilwong

Differential Revision: D65960050

fbshipit-source-id: d647ef90b45d9e930310cdcba159159d8e773213
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