[quantization] Add some support for 3d operations#50003
[quantization] Add some support for 3d operations#50003dmenig wants to merge 3 commits intopytorch:masterfrom
Conversation
|
Hi @hyperfraise! Thank you for your pull request and welcome to our community. We require contributors to sign our Contributor License Agreement, and we don't seem to have you on file. In order for us to review and merge your code, please sign at https://code.facebook.com/cla. If you are contributing on behalf of someone else (eg your employer), the individual CLA may not be sufficient and your employer may need to sign the corporate CLA. If you have received this in error or have any questions, please contact us at cla@fb.com. Thanks! |
💊 CI failures summary and remediationsAs of commit c11dd20 (more details on the Dr. CI page):
🚧 1 fixed upstream failure:These were probably caused by upstream breakages that were already fixed.
Please rebase on the
|
57b6b3e to
39ad9a4
Compare
3f03c28 to
5b7543f
Compare
Codecov Report
@@ Coverage Diff @@
## master #50003 +/- ##
==========================================
- Coverage 80.80% 80.59% -0.22%
==========================================
Files 1972 1957 -15
Lines 216286 214523 -1763
==========================================
- Hits 174774 172888 -1886
- Misses 41512 41635 +123 |
|
Could you separate the formatting changes and the feature? |
|
can you update the test here: https://github.com/pytorch/pytorch/blob/master/test/quantization/test_quantize_fx.py#L397 (add a 3 to the list) |
Thanks for the review ! They are separated in different commits; Do you mean opening a new PR ? |
I will. |
I see. ideally it should be a new PR I think, but separate commits probably works as well. |
jerryzh168
left a comment
There was a problem hiding this comment.
Looks good, thanks for adding this!
Wait there's the rocm test I can't pass, do you see why ? I can't figure why |
jerryzh168
left a comment
There was a problem hiding this comment.
you will need to update L337
"call_function", torch.nn.functional.conv2d, args, kwargs
of torch/quantization/fx/quantization_patterns.py as well I think.
could you also add a test for conv3d in https://github.com/pytorch/pytorch/blob/master/test/quantization/test_quantize_fx.py#L233?
7972d9e to
1a8701a
Compare
|
@albanD @mingzhe09088 @mrshenli @pritamdamania87 @rohan-varma @zhaojuanmao @jerryzh168 @ezyang @fmassa @glaringlee @soumith |
facebook-github-bot
left a comment
There was a problem hiding this comment.
@jerryzh168 has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
0ae48a8 to
e431ba5
Compare
facebook-github-bot
left a comment
There was a problem hiding this comment.
@jerryzh168 has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
0b42d0f to
1f1d4f4
Compare
|
cc @jerryzh168 Is this PR ready to land? |
facebook-github-bot
left a comment
There was a problem hiding this comment.
@jerryzh168 has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
|
Sorry @hyperfraise for the late response, there were still some internal errors (in phabricator) in aten/src/ATen/native/metal/MetalPrepackOpRegister.cpp: cc @xta0 is there a way that we can compile metal in OSS to make sure it works? |
|
Hi @hyperfraise could you revert the metal changes? we can add it internal if it's needed. the ci/clang errors are only visible internally so it's hard to figure out what to do in oss. |
Ofc ! Let's drop them. |
|
I rebased with your suggestions. It seems there's an unrelated failure in the CI, so you can still review. |
facebook-github-bot
left a comment
There was a problem hiding this comment.
@jerryzh168 has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
|
@jerryzh168 merged this pull request in f918597. |
|
@hyperfraise sorry for the long delay, it's finally landed! |
Nothing undeserved ^^ Glad to open the next PR if the issue mandates one. Thanks for your help. |
Fixes #50002
The last commit adds tests for 3d conv with the
SubModelFusionandSubModelWithoutFusionclasses.