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
all_to_one cuda support non-2d inputs #2575
base: main
Are you sure you want to change the base?
Conversation
This pull request was exported from Phabricator. Differential Revision: D57180617 |
✅ Deploy Preview for pytorch-fbgemm-docs ready!
To edit notification comments on pull requests, go to your Netlify site configuration. |
Summary: Support non-2d tensors in all_to_one. Sometimes torchrec outputs are empty tensors. This can be solved on python module side guarding empty tensors, but we d better not to have shape dependent logic for PT2 tracing. Differential Revision: D57180617
f8f7c44
to
01d742c
Compare
This pull request was exported from Phabricator. Differential Revision: D57180617 |
Summary: Support non-2d tensors in all_to_one. Sometimes torchrec outputs are empty tensors. This can be solved on python module side guarding empty tensors, but we d better not to have shape dependent logic for PT2 tracing. Differential Revision: D57180617
01d742c
to
316aacb
Compare
This pull request was exported from Phabricator. Differential Revision: D57180617 |
Summary: Support non-2d tensors in all_to_one. Sometimes torchrec outputs are empty tensors. This can be solved on python module side guarding empty tensors, but we d better not to have shape dependent logic for PT2 tracing. Differential Revision: D57180617
316aacb
to
2e5438a
Compare
This pull request was exported from Phabricator. Differential Revision: D57180617 |
Summary:
Support non-2d tensors in all_to_one.
Sometimes torchrec outputs are empty tensors.
This can be solved on python module side guarding empty tensors, but we d better not to have shape dependent logic for PT2 tracing.
Differential Revision: D57180617