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

support permute_multi_embedding_function on torch.export #3897

Closed
wants to merge 1 commit into from

Conversation

zejunh
Copy link
Contributor

@zejunh zejunh commented Mar 28, 2025

Summary:
support fbgemm.permute_multi_embedding_function.default for LPV model
register with separate entry in kernel.yaml for graph mode lowering
added fp16 ref kernel (sitecao)

Differential Revision: D71821354

Copy link

netlify bot commented Mar 28, 2025

Deploy Preview for pytorch-fbgemm-docs ready!

Name Link
🔨 Latest commit bec1e54
🔍 Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/67f06a4ed6f7b100082ceadb
😎 Deploy Preview https://deploy-preview-3897--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: D71821354

@facebook-github-bot
Copy link
Contributor

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

zejunh added a commit to zejunh/FBGEMM that referenced this pull request Mar 28, 2025
Summary:
X-link: facebookresearch/FBGEMM#988

Pull Request resolved: pytorch#3897

support fbgemm.permute_multi_embedding_function.default for LPV model
register with separate entry in kernel.yaml for graph mode lowering
added fp16 ref kernel (sitecao)

Differential Revision: D71821354
@zejunh zejunh force-pushed the export-D71821354 branch from 2f4ed11 to 4d1418d Compare March 28, 2025 22:47
@facebook-github-bot
Copy link
Contributor

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

zejunh added a commit to zejunh/FBGEMM that referenced this pull request Mar 28, 2025
Summary:
X-link: facebookresearch/FBGEMM#988

Pull Request resolved: pytorch#3897

support fbgemm.permute_multi_embedding_function.default for LPV model
register with separate entry in kernel.yaml for graph mode lowering
added fp16 ref kernel (sitecao)

Differential Revision: D71821354
@zejunh zejunh force-pushed the export-D71821354 branch from 4d1418d to de278b0 Compare March 28, 2025 22:53
@facebook-github-bot
Copy link
Contributor

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

zejunh added a commit to zejunh/FBGEMM that referenced this pull request Mar 31, 2025
Summary:
X-link: facebookresearch/FBGEMM#988

Pull Request resolved: pytorch#3897

support fbgemm.permute_multi_embedding_function.default for LPV model
register with separate entry in kernel.yaml for graph mode lowering
added fp16 ref kernel (sitecao)

Differential Revision: D71821354
@zejunh zejunh force-pushed the export-D71821354 branch from a39c7cc to b970886 Compare March 31, 2025 20:54
@facebook-github-bot
Copy link
Contributor

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

zejunh added a commit to zejunh/FBGEMM that referenced this pull request Mar 31, 2025
Summary:
X-link: facebookresearch/FBGEMM#988

Pull Request resolved: pytorch#3897

support fbgemm.permute_multi_embedding_function.default for LPV model
register with separate entry in kernel.yaml for graph mode lowering
added fp16 ref kernel (sitecao)

Differential Revision: D71821354
@zejunh zejunh force-pushed the export-D71821354 branch from b970886 to 3b3fca5 Compare March 31, 2025 21:01
@facebook-github-bot
Copy link
Contributor

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

zejunh added a commit to zejunh/FBGEMM that referenced this pull request Mar 31, 2025
Summary:
X-link: facebookresearch/FBGEMM#988

Pull Request resolved: pytorch#3897

support fbgemm.permute_multi_embedding_function.default for LPV model
register with separate entry in kernel.yaml for graph mode lowering
added fp16 ref kernel (sitecao)

Differential Revision: D71821354
@zejunh zejunh force-pushed the export-D71821354 branch from 3b3fca5 to 5dfc705 Compare March 31, 2025 21:05
@facebook-github-bot
Copy link
Contributor

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

@zejunh zejunh force-pushed the export-D71821354 branch from 6f1f63f to cf99c2b Compare April 3, 2025 00:42
zejunh added a commit to zejunh/FBGEMM that referenced this pull request Apr 3, 2025
Summary:
X-link: facebookresearch/FBGEMM#988

Pull Request resolved: pytorch#3897

support fbgemm.permute_multi_embedding_function.default for LPV model
register with separate entry in kernel.yaml for graph mode lowering
added fp16 ref kernel (sitecao)

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

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

@zejunh zejunh force-pushed the export-D71821354 branch from cf99c2b to f738e20 Compare April 3, 2025 00:50
zejunh added a commit to zejunh/FBGEMM that referenced this pull request Apr 3, 2025
Summary:
X-link: facebookresearch/FBGEMM#988

Pull Request resolved: pytorch#3897

support fbgemm.permute_multi_embedding_function.default for LPV model
register with separate entry in kernel.yaml for graph mode lowering
added fp16 ref kernel (sitecao)

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

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

@zejunh zejunh force-pushed the export-D71821354 branch from f738e20 to 50bcd99 Compare April 4, 2025 19:15
zejunh added a commit to zejunh/FBGEMM that referenced this pull request Apr 4, 2025
Summary:
X-link: facebookresearch/FBGEMM#988

Pull Request resolved: pytorch#3897

support fbgemm.permute_multi_embedding_function.default for LPV model
register with separate entry in kernel.yaml for graph mode lowering
added fp16 ref kernel (sitecao)

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

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

zejunh added a commit to zejunh/FBGEMM that referenced this pull request Apr 4, 2025
Summary:
X-link: facebookresearch/FBGEMM#988

Pull Request resolved: pytorch#3897

support fbgemm.permute_multi_embedding_function.default for LPV model
register with separate entry in kernel.yaml for graph mode lowering
added fp16 ref kernel (sitecao)

Differential Revision: D71821354
@zejunh zejunh force-pushed the export-D71821354 branch from 50bcd99 to 79ff1dc Compare April 4, 2025 19:23
@facebook-github-bot
Copy link
Contributor

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

zejunh added a commit to zejunh/FBGEMM that referenced this pull request Apr 4, 2025
Summary:
X-link: facebookresearch/FBGEMM#988

Pull Request resolved: pytorch#3897

support fbgemm.permute_multi_embedding_function.default for LPV model
register with separate entry in kernel.yaml for graph mode lowering
added fp16 ref kernel (sitecao)

Differential Revision: D71821354
@zejunh zejunh force-pushed the export-D71821354 branch from 79ff1dc to 87e794a Compare April 4, 2025 20:09
@facebook-github-bot
Copy link
Contributor

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

zejunh added a commit to zejunh/FBGEMM that referenced this pull request Apr 4, 2025
Summary:
X-link: facebookresearch/FBGEMM#988

Pull Request resolved: pytorch#3897

support fbgemm.permute_multi_embedding_function.default for LPV model
register with separate entry in kernel.yaml for graph mode lowering
added fp16 ref kernel (sitecao)

Differential Revision: D71821354
@zejunh zejunh force-pushed the export-D71821354 branch from 87e794a to fd26678 Compare April 4, 2025 20:16
@facebook-github-bot
Copy link
Contributor

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

zejunh added a commit to zejunh/FBGEMM that referenced this pull request Apr 4, 2025
Summary:
X-link: facebookresearch/FBGEMM#988

Pull Request resolved: pytorch#3897

support fbgemm.permute_multi_embedding_function.default for LPV model
register with separate entry in kernel.yaml for graph mode lowering
added fp16 ref kernel (sitecao)

Differential Revision: D71821354
@zejunh zejunh force-pushed the export-D71821354 branch from fd26678 to 71dfe7e Compare April 4, 2025 20:27
@facebook-github-bot
Copy link
Contributor

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

@zejunh zejunh force-pushed the export-D71821354 branch from 71dfe7e to 4c1e502 Compare April 4, 2025 23:16
zejunh added a commit to zejunh/FBGEMM that referenced this pull request Apr 4, 2025
Summary:
X-link: facebookresearch/FBGEMM#988

Pull Request resolved: pytorch#3897

support fbgemm.permute_multi_embedding_function.default for LPV model
register with separate entry in kernel.yaml for graph mode lowering
added fp16 ref kernel (sitecao)

Differential Revision: D71821354
Summary:
X-link: facebookresearch/FBGEMM#988

Pull Request resolved: pytorch#3897

support fbgemm.permute_multi_embedding_function.default for LPV model
register with separate entry in kernel.yaml for graph mode lowering
added fp16 ref kernel (sitecao)

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

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

@facebook-github-bot
Copy link
Contributor

This pull request has been merged in 3c1f215.

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

X-link: pytorch#3897

support fbgemm.permute_multi_embedding_function.default for LPV model
register with separate entry in kernel.yaml for graph mode lowering
added fp16 ref kernel (sitecao)

Reviewed By: StellarrZ

Differential Revision: D71821354

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