Use PyTorch dispatcher in TBE training GPU autograds #1948
Closed
+776
−463
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
✅ Deploy Preview for pytorch-fbgemm-docs canceled.
|
|
This pull request was exported from Phabricator. Differential Revision: D48383830 |
|
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
force-pushed
the
export-D48383830
branch
from
2b5af62 to
81d8b41
Compare
|
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
force-pushed
the
export-D48383830
branch
from
81d8b41 to
f1855af
Compare
|
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
force-pushed
the
export-D48383830
branch
from
f1855af to
9d6f3a5
Compare
|
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
force-pushed
the
export-D48383830
branch
from
9d6f3a5 to
b1d98e1
Compare
|
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
force-pushed
the
export-D48383830
branch
from
b1d98e1 to
dde7c20
Compare
|
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
force-pushed
the
export-D48383830
branch
from
dde7c20 to
25bd111
Compare
|
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
force-pushed
the
export-D48383830
branch
from
25bd111 to
df35664
Compare
|
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
force-pushed
the
export-D48383830
branch
from
df35664 to
943190e
Compare
|
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
force-pushed
the
export-D48383830
branch
from
943190e to
ff6e421
Compare
|
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
force-pushed
the
export-D48383830
branch
from
ff6e421 to
2f390ab
Compare
|
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
force-pushed
the
export-D48383830
branch
from
2f390ab to
19fc46e
Compare
|
This pull request was exported from Phabricator. Differential Revision: D48383830 |
1 similar comment
|
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
force-pushed
the
export-D48383830
branch
from
19fc46e to
5e6a9e9
Compare
|
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
force-pushed
the
export-D48383830
branch
from
5e6a9e9 to
61c0272
Compare
|
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
force-pushed
the
export-D48383830
branch
from
61c0272 to
9c96b4d
Compare
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
|
This pull request was exported from Phabricator. Differential Revision: D48383830 |
sryap
force-pushed
the
export-D48383830
branch
from
9c96b4d to
103ec71
Compare
|
This pull request has been merged in 5a6cc0e. |
Jason-KChen
pushed a commit
to Jason-KChen/FBGEMM
that referenced
this pull request
Sep 23, 2025
Summary: Pull Request resolved: facebookresearch/FBGEMM#1948 Differential Revision: D83100663
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
2 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Summary:
Register TBE GPU training operators in PyTorch and invoke them in
autograd via PyTorch dispatcher
Differential Revision: D48383830