bug fix 19374 - fix for upsample export#20116
Conversation
|
bug fix #19374 |
250dfe6 to
d7faf52
Compare
houseroad
left a comment
There was a problem hiding this comment.
Looks good, can you add a test case to cover the code you add?
Here is an example: https://github.com/pytorch/pytorch/blob/master/test/onnx/test_pytorch_onnx_caffe2.py#L958
d7faf52 to
5688c7e
Compare
Sure, thanks for the reference. Added the test to cover new code. |
facebook-github-bot
left a comment
There was a problem hiding this comment.
@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
5688c7e to
d21c1e7
Compare
facebook-github-bot
left a comment
There was a problem hiding this comment.
@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
|
We should enable the test for Upsample in test_operators.py also. pytorch/test/onnx/test_operators.py Line 476 in 4d676d5 |
|
torch/onnx/symbolic.py
Outdated
There was a problem hiding this comment.
I guess there has been updates to upsample that for the argument output_size it could be either a tensor or a int list. Thus the annotation becomes @parse_args('v', 'v') which can be ignored.
There was a problem hiding this comment.
That's right, as @BowenBao mentioned adding @pars_args('v','v') will be redundant.
d21c1e7 to
4c75ed8
Compare
@Alcaster Good point, we should definitely address other methods as well. But since in this PR we are trying to fix and unblock issue #19374 (which uses 'nearest' method), how about addressing other methods in a separate PR ? |
|
The failing onnx ci is legit. Please fix the failing tests, thanks. |
|
Does this also fix the "bilinear" upsampling export? |
facebook-github-bot
left a comment
There was a problem hiding this comment.
@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
|
@pk-g friendly ping for fix :-) |
|
Pls fix =) |
4c75ed8 to
60ae121
Compare
60ae121 to
1867cd0
Compare
facebook-github-bot
left a comment
There was a problem hiding this comment.
@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
|
@pk-g please rebase again to make sure the CI pass. The broken CI has been fixed already. Thanks! |
1867cd0 to
d7010eb
Compare
facebook-github-bot
left a comment
There was a problem hiding this comment.
@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
houseroad
left a comment
There was a problem hiding this comment.
LGTM, thanks for fixing the problem.
|
@houseroad merged this pull request in 93d5503. |
No description provided.