Skip to content

Add meta function for PT2 wrappers #3240

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

spcyppt
Copy link
Contributor

@spcyppt spcyppt commented Oct 11, 2024

Summary:
X-link: https://github.com/facebookresearch/FBGEMM/pull/341

To be pt2 compliant, add meta functions for PT2 wrappers and dispatch the ops to META.

Without meta dispatch, pt2 tests will fail on torch.compile with error like:

Unsupported: Backend compiler failed with a fake tensor exception
...
Caused by UnsupportedOperatorException: fbgemm.split_embedding_backward_codegen_sgd_unweighted_vbe_pt2_wrapper.default

Differential Revision: D64255747

@facebook-github-bot
Copy link
Contributor

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

Copy link

netlify bot commented Oct 11, 2024

Deploy Preview for pytorch-fbgemm-docs ready!

Name Link
🔨 Latest commit 97a23bd
🔍 Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/670c3f8e8f69300008b61485
😎 Deploy Preview https://deploy-preview-3240--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.

spcyppt added a commit to spcyppt/FBGEMM that referenced this pull request Oct 11, 2024
Summary:

X-link: facebookresearch/FBGEMM#341

To be pt2 compliant, add meta functions for PT2 wrappers and dispatch the ops to META.

Without meta dispatch, pt2 tests will fail on `torch.compile` with error like:
```
Unsupported: Backend compiler failed with a fake tensor exception
...
Caused by UnsupportedOperatorException: fbgemm.split_embedding_backward_codegen_sgd_unweighted_vbe_pt2_wrapper.default
```

Differential Revision: D64255747
spcyppt added a commit to spcyppt/FBGEMM that referenced this pull request Oct 12, 2024
Summary:

X-link: facebookresearch/FBGEMM#341

To be pt2 compliant, add meta functions for PT2 wrappers and dispatch the ops to META.

Without meta dispatch, pt2 tests will fail on `torch.compile` with error like:
```
Unsupported: Backend compiler failed with a fake tensor exception
...
Caused by UnsupportedOperatorException: fbgemm.split_embedding_backward_codegen_sgd_unweighted_vbe_pt2_wrapper.default
```

Reviewed By: q10

Differential Revision: D64255747
Summary:

X-link: facebookresearch/FBGEMM#341

To be pt2 compliant, add meta functions for PT2 wrappers and dispatch the ops to META.

Without meta dispatch, pt2 tests will fail on `torch.compile` with error like:
```
Unsupported: Backend compiler failed with a fake tensor exception
...
Caused by UnsupportedOperatorException: fbgemm.split_embedding_backward_codegen_sgd_unweighted_vbe_pt2_wrapper.default
```

Reviewed By: q10

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

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

@facebook-github-bot
Copy link
Contributor

This pull request has been merged in c8e32e4.

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

Pull Request resolved: facebookresearch/FBGEMM#341

To be pt2 compliant, add meta functions for PT2 wrappers and dispatch the ops to META.

Without meta dispatch, pt2 tests will fail on `torch.compile` with error like:
```
Unsupported: Backend compiler failed with a fake tensor exception
...
Caused by UnsupportedOperatorException: fbgemm.split_embedding_backward_codegen_sgd_unweighted_vbe_pt2_wrapper.default
```

Reviewed By: q10

Differential Revision: D64255747

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