-
Notifications
You must be signed in to change notification settings - Fork 25.6k
Add the fusion of quantized batchnorm and relu #34795
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
Conversation
fix the test Add the batch_norm3d and batch_norm3d_relu fused quantized op fix linter make the type be consistent Add the fusion of quantized batchnorm and relu
💊 CircleCI build failures summary and remediationsAs of commit 3554ef6 (more details on the Dr. CI page):
🕵️ 3 new failures recognized by patternsThe following build failures do not appear to be due to upstream breakages:
|
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.
@lly-zero-one has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
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.
@lly-zero-one has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
Do we not need changes in the python side to do the fusion of these ops as well? |
Yes, I have another PR in my sandbox. It will be together with conv3d and the quantization flow change. |
Can we merge this PR with the one that has the test for it? I'm wary of enabling fused Relu without testing it. |
I am adding a bn3drelu test to this PR. |
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.
@lly-zero-one has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
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
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.
@lly-zero-one is landing this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
Unlanding. This appears to have broken the the Mac build:
|
@lly-zero-one merged this pull request in 733b631. |
As title, we want to support the BN2d_relu and BN3d_relu
Test to be added!