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

Fix get_unique_indices_v2 registration #2993

Closed
wants to merge 1 commit into from

Conversation

sryap
Copy link
Contributor

@sryap sryap commented Aug 14, 2024

Summary:
Properly register get_unique_indices_v2 in torch.ops.fbgemm and
import it in SSD-TBE.

  • Use init to define and register each op only once
  • Check if an op is defined before defining and registering it to
    avoid duplicate declaration

Differential Revision: D61294287

Copy link

netlify bot commented Aug 14, 2024

Deploy Preview for pytorch-fbgemm-docs ready!

Name Link
🔨 Latest commit 94e7157
🔍 Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/66be81690165b10008014386
😎 Deploy Preview https://deploy-preview-2993--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.

@facebook-github-bot
Copy link
Contributor

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

@facebook-github-bot
Copy link
Contributor

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

@sryap sryap force-pushed the export-D61294287 branch from 7391569 to 0d89feb Compare August 15, 2024 03:32
@facebook-github-bot
Copy link
Contributor

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

@sryap sryap force-pushed the export-D61294287 branch from 0d89feb to 720f379 Compare August 15, 2024 03:37
sryap added a commit to sryap/FBGEMM that referenced this pull request Aug 15, 2024
Summary:
X-link: facebookresearch/FBGEMM#85

Pull Request resolved: pytorch#2993

Properly register `get_unique_indices_v2` in `torch.ops.fbgemm` and
import it in SSD-TBE.
- Use `init` to define and register each op only once
- Check if an op is defined before defining and registering it to
  avoid duplicate declaration

Reviewed By: q10

Differential Revision: D61294287
Summary:
X-link: facebookresearch/FBGEMM#85

Pull Request resolved: pytorch#2993

Remove `get_unique_indices_v2` registration in `torch.ops.fbgemm` as
we cannot cleanly/safely register it

Reviewed By: q10

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

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

@sryap sryap force-pushed the export-D61294287 branch from 720f379 to 94e7157 Compare August 15, 2024 22:30
sryap added a commit to sryap/FBGEMM that referenced this pull request Aug 15, 2024
Summary:
X-link: facebookresearch/FBGEMM#85

Pull Request resolved: pytorch#2993

Remove `get_unique_indices_v2` registration in `torch.ops.fbgemm` as
we cannot cleanly/safely register it

Reviewed By: q10

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

This pull request has been merged in 160633c.

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