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

batched_dense_vec_jagged_2d_mul add Meta and Autograd backend #1468

Closed
wants to merge 5 commits into from

Conversation

brad-mengchi
Copy link
Contributor

Summary:
Add Meta and Autograd backend for batched_dense_vec_jagged_2d_mul for dynamo and AOT autograd.

A more proper way of doing Autograd to make inductor working for jagged ops. Also added Meta tensors so we're not mercy of arbitrary zero inputs fed into by default.

Differential Revision: D41387994

@netlify
Copy link

netlify bot commented Nov 18, 2022

Deploy Preview for pytorch-fbgemm-docs canceled.

Name Link
🔨 Latest commit 36cc29a
🔍 Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/637dc1b403c435000b4e6777

@facebook-github-bot
Copy link
Contributor

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

jianyuh and others added 5 commits November 22, 2022 22:45
Summary:
Pull Request resolved: pytorch#1472

As title

Reviewed By: brad-mengchi, jiawenliu64

Differential Revision: D41445777

fbshipit-source-id: 510e30e2292b617944ebb7a7d2fe51f0a544f529
Differential Revision: https://internalfb.com/D41367064

fbshipit-source-id: 6f2a7b923dc337548d87aa2828e431dab40bc706
Differential Revision: D41368350

fbshipit-source-id: 21519b9b4017ed480217011bee74051c465f6e03
Differential Revision: D41383597

fbshipit-source-id: 7aeed19578feb3ceffbac3a28c562222123b195f
…h#1468)

Summary:
Pull Request resolved: pytorch#1468

Add Meta and Autograd backend for batched_dense_vec_jagged_2d_mul for dynamo and AOT autograd.

A more proper way of doing Autograd to make inductor working for jagged ops. Also added Meta tensors so we're not mercy of arbitrary zero inputs fed into by default.

Differential Revision: D41387994

fbshipit-source-id: 827d2fb26c8c173c4af9bb52b7fadf75ded1105b
@facebook-github-bot
Copy link
Contributor

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

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.

3 participants