Fill eigenvector with zeros if not required#10645
Fill eigenvector with zeros if not required#10645ailzhang wants to merge 4 commits intopytorch:masterfrom
Conversation
|
I think it would be nice to have a test for this. |
facebook-github-bot
left a comment
There was a problem hiding this comment.
ailzhang has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
li-roy
left a comment
There was a problem hiding this comment.
Should we consider adding python wrapper and just returning a single Tensor in the eigenvectors=False case? Other than that, lgtm.
test/test_torch.py
Outdated
| resv2 = torch.randn(3, 3) | ||
| cov = cov.cuda() | ||
| rese2 = torch.zeros(3).cuda() | ||
| resv2 = torch.randn(3, 3).cuda() |
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
facebook-github-bot
left a comment
There was a problem hiding this comment.
ailzhang has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
facebook-github-bot
left a comment
There was a problem hiding this comment.
ailzhang has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
|
@pytorchbot retest this please |
1 similar comment
|
@pytorchbot retest this please |
Summary: Fix #10345, which only happens in CUDA case. * Instead of returning some random buffer, we fill it with zeros. * update torch.symeig doc. Pull Request resolved: pytorch/pytorch#10645 Reviewed By: soumith Differential Revision: D9395762 Pulled By: ailzhang fbshipit-source-id: 0f3ed9bb6a919a9c1a4b8eb45188f65a68bfa9ba
Summary: Fix pytorch#10345, which only happens in CUDA case. * Instead of returning some random buffer, we fill it with zeros. * update torch.symeig doc. Pull Request resolved: pytorch#10645 Reviewed By: soumith Differential Revision: D9395762 Pulled By: ailzhang fbshipit-source-id: 0f3ed9bb6a919a9c1a4b8eb45188f65a68bfa9ba
Fix #10345, which only happens in CUDA case.
Instead of returning some random buffer, we fill it with zeros.
update torch.symeig doc.