-
Notifications
You must be signed in to change notification settings - Fork 26.3k
[ONNX] Export det #26958
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
[ONNX] Export det #26958
Conversation
BowenBao
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM, minor comment on better test coverage.
| def forward(self, x): | ||
| return torch.det(x) | ||
|
|
||
| x = torch.randn(2, 3, 5, 5, requires_grad=True) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Can we add another test case where input shape is (5, 5)? That covers the case where output is scalar.
|
@neginraoof TestOperators.test_det is failing |
|
@pytorchbot rebase this please |
houseroad
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Export looks good.
RuntimeError: lu: LAPACK library not found in compilation
Please fix the tests.
facebook-github-bot
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
|
@houseroad I didn't see this locally. Trying to repro. Do you have an idea why it complains about the missing package? |
|
cc @houseroad for review |
houseroad
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Fix the conflict?
|
@houseroad just updated. please review |
houseroad
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please do NOT update onnx submodule
|
@houseroad onnx is reset. Just rebased this |
|
@houseroad this is ready |
|
@houseroad please review the updates |
houseroad
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
facebook-github-bot
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
|
@houseroad merged this pull request in 4f70b5a. |
Summary: Added symbolic to export det in opset 11 Updating ONNX submodule is required for det export Pull Request resolved: pytorch#26958 Reviewed By: hl475 Differential Revision: D17844887 Pulled By: houseroad fbshipit-source-id: 224ae3ff82939dc7ae8584c5a30a31fe6afa05f6
Added symbolic to export det in opset 11
Updating ONNX submodule is required for det export