Skip to content
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

Use PyTorch dispatcher in TBE training GPU autograds #1948

Closed
wants to merge 1 commit into from

Conversation

sryap
Copy link
Contributor

@sryap sryap commented Aug 17, 2023

Summary:
Register TBE GPU training operators in PyTorch and invoke them in
autograd via PyTorch dispatcher

Differential Revision: D48383830

@netlify
Copy link

netlify bot commented Aug 17, 2023

Deploy Preview for pytorch-fbgemm-docs canceled.

Name Link
🔨 Latest commit 103ec71
🔍 Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/64e8543758800700087eb206

@facebook-github-bot
Copy link
Contributor

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

@facebook-github-bot
Copy link
Contributor

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

sryap added a commit to sryap/FBGEMM that referenced this pull request Aug 17, 2023
Summary:
Pull Request resolved: pytorch#1948

Register TBE GPU training operators in PyTorch and invoke them in
autograd via PyTorch dispatcher

Differential Revision: D48383830

fbshipit-source-id: 78adce13952a89d215332a61d88fe2a40f8b0d3e
@sryap sryap force-pushed the export-D48383830 branch from 2b5af62 to 81d8b41 Compare August 17, 2023 01:22
@facebook-github-bot
Copy link
Contributor

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

sryap added a commit to sryap/FBGEMM that referenced this pull request Aug 17, 2023
Summary:
Pull Request resolved: pytorch#1948

Register TBE GPU training operators in PyTorch and invoke them in
autograd via PyTorch dispatcher

Differential Revision: D48383830

fbshipit-source-id: d61548be67b33629772099631501626f2594ea4e
@sryap sryap force-pushed the export-D48383830 branch from 81d8b41 to f1855af Compare August 17, 2023 01:29
@facebook-github-bot
Copy link
Contributor

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

sryap added a commit to sryap/FBGEMM that referenced this pull request Aug 17, 2023
Summary:
Pull Request resolved: pytorch#1948

Register TBE GPU training operators in PyTorch and invoke them in
autograd via PyTorch dispatcher

Differential Revision: D48383830

fbshipit-source-id: cefe5da590956d07f369a80709d3cb0b2745b014
@sryap sryap force-pushed the export-D48383830 branch from f1855af to 9d6f3a5 Compare August 17, 2023 02:33
@facebook-github-bot
Copy link
Contributor

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

sryap added a commit to sryap/FBGEMM that referenced this pull request Aug 17, 2023
Summary:
Pull Request resolved: pytorch#1948

Register TBE GPU training operators in PyTorch and invoke them in
autograd via PyTorch dispatcher

Differential Revision: D48383830

fbshipit-source-id: 706263b4e8b9df63ee7dd405198401957b63e10c
@sryap sryap force-pushed the export-D48383830 branch from 9d6f3a5 to b1d98e1 Compare August 17, 2023 02:40
@facebook-github-bot
Copy link
Contributor

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

sryap added a commit to sryap/FBGEMM that referenced this pull request Aug 17, 2023
Summary:
Pull Request resolved: pytorch#1948

Register TBE GPU training operators in PyTorch and invoke them in
autograd via PyTorch dispatcher

Differential Revision: D48383830

fbshipit-source-id: 3fc408c0d48b68a392efa05526298d8dbc6d1c4d
@sryap sryap force-pushed the export-D48383830 branch from b1d98e1 to dde7c20 Compare August 17, 2023 06:24
@facebook-github-bot
Copy link
Contributor

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

sryap added a commit to sryap/FBGEMM that referenced this pull request Aug 17, 2023
Summary:
Pull Request resolved: pytorch#1948

Register TBE GPU training operators in PyTorch and invoke them in
autograd via PyTorch dispatcher

Differential Revision: D48383830

fbshipit-source-id: 37157495c820fa3a3fb34fe28765877cf88bbb79
@sryap sryap force-pushed the export-D48383830 branch from dde7c20 to 25bd111 Compare August 17, 2023 06:31
@facebook-github-bot
Copy link
Contributor

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

sryap added a commit to sryap/FBGEMM that referenced this pull request Aug 19, 2023
Summary:
Pull Request resolved: pytorch#1948

Register TBE GPU training operators in PyTorch and invoke them in
autograd via PyTorch dispatcher

Differential Revision: D48383830

fbshipit-source-id: 620b655eda099c6a2c35dd48df63567d2b474e9d
@sryap sryap force-pushed the export-D48383830 branch from 25bd111 to df35664 Compare August 19, 2023 01:34
@facebook-github-bot
Copy link
Contributor

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

sryap added a commit to sryap/FBGEMM that referenced this pull request Aug 19, 2023
Summary:
Pull Request resolved: pytorch#1948

Register TBE GPU training operators in PyTorch and invoke them in
autograd via PyTorch dispatcher

Differential Revision: D48383830

fbshipit-source-id: 365955a61ef79672fb47370370473a96f3a45a38
@sryap sryap force-pushed the export-D48383830 branch from df35664 to 943190e Compare August 19, 2023 01:41
@facebook-github-bot
Copy link
Contributor

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

sryap added a commit to sryap/FBGEMM that referenced this pull request Aug 24, 2023
Summary:
Pull Request resolved: pytorch#1948

Register TBE GPU training operators in PyTorch and invoke them in
autograd via PyTorch dispatcher

Reviewed By: jianyuh

Differential Revision: D48383830

fbshipit-source-id: 6ec39bb08b7f25b44e464f52801ddd171676c5a6
@sryap sryap force-pushed the export-D48383830 branch from 943190e to ff6e421 Compare August 24, 2023 04:46
@facebook-github-bot
Copy link
Contributor

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

sryap added a commit to sryap/FBGEMM that referenced this pull request Aug 24, 2023
Summary:
Pull Request resolved: pytorch#1948

Register TBE GPU training operators in PyTorch and invoke them in
autograd via PyTorch dispatcher

Reviewed By: jianyuh

Differential Revision: D48383830

fbshipit-source-id: 274f1a41c4600b4cb6e9d308add8df8fe2ec7ee0
@sryap sryap force-pushed the export-D48383830 branch from ff6e421 to 2f390ab Compare August 24, 2023 04:51
@facebook-github-bot
Copy link
Contributor

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

sryap added a commit to sryap/FBGEMM that referenced this pull request Aug 24, 2023
Summary:
Pull Request resolved: pytorch#1948

Register TBE GPU training operators in PyTorch and invoke them in
autograd via PyTorch dispatcher

Reviewed By: jianyuh

Differential Revision: D48383830

fbshipit-source-id: cb5e81ffbd2a46cfe7b723653e031e48c8b2b19e
@sryap sryap force-pushed the export-D48383830 branch from 2f390ab to 19fc46e Compare August 24, 2023 19:04
@facebook-github-bot
Copy link
Contributor

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

1 similar comment
@facebook-github-bot
Copy link
Contributor

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

sryap added a commit to sryap/FBGEMM that referenced this pull request Aug 25, 2023
Summary:
Pull Request resolved: pytorch#1948

Register TBE GPU training operators in PyTorch and invoke them in
autograd via PyTorch dispatcher

Reviewed By: jianyuh

Differential Revision: D48383830

fbshipit-source-id: 185aa4b600672a3850f8545a1b46a542a424b8b2
@sryap sryap force-pushed the export-D48383830 branch from 19fc46e to 5e6a9e9 Compare August 25, 2023 05:54
@facebook-github-bot
Copy link
Contributor

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

sryap added a commit to sryap/FBGEMM that referenced this pull request Aug 25, 2023
Summary:
Pull Request resolved: pytorch#1948

Register TBE GPU training operators in PyTorch and invoke them in
autograd via PyTorch dispatcher

Reviewed By: jianyuh

Differential Revision: D48383830

fbshipit-source-id: 125e19e1886a6a318a443805451e55c28203d28e
@sryap sryap force-pushed the export-D48383830 branch from 5e6a9e9 to 61c0272 Compare August 25, 2023 06:02
@facebook-github-bot
Copy link
Contributor

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

sryap added a commit to sryap/FBGEMM that referenced this pull request Aug 25, 2023
Summary:
Pull Request resolved: pytorch#1948

Register TBE GPU training operators in PyTorch and invoke them in
autograd via PyTorch dispatcher

Reviewed By: jianyuh

Differential Revision: D48383830

fbshipit-source-id: b1f7fb274d17366a2b7db55227aaab41d9ba2eb8
@sryap sryap force-pushed the export-D48383830 branch from 61c0272 to 9c96b4d Compare August 25, 2023 07:05
Summary:
Pull Request resolved: pytorch#1948

Register TBE GPU training operators in PyTorch and invoke them in
autograd via PyTorch dispatcher

Reviewed By: jianyuh

Differential Revision: D48383830

fbshipit-source-id: 789b83576092708a68440a6c5311a1ef4c50698c
@facebook-github-bot
Copy link
Contributor

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

@facebook-github-bot
Copy link
Contributor

This pull request has been merged in 5a6cc0e.

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