-
Notifications
You must be signed in to change notification settings - Fork 25.5k
[quant][graphmode][fx] Add support for conv module #43285
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
Summary: Porting op tests from test_quantize_jit.py Test Plan: TestQuantizeFxOps Reviewers: Subscribers: Tasks: Tags: [ghstack-poisoned]
💊 CI failures summary and remediationsAs of commit dabe528 (more details on the Dr. CI page): ✅ None of the CI failures appear to be your fault 💚
🚧 1 fixed upstream failure:These were probably caused by upstream breakages that were already fixed.
Please rebase on the
|
Summary: Porting op tests from test_quantize_jit.py Test Plan: TestQuantizeFxOps Reviewers: Subscribers: Tasks: Tags: [ghstack-poisoned]
Summary: Porting op tests from test_quantize_jit.py Test Plan: TestQuantizeFxOps Reviewers: Subscribers: Tasks: Tags: Differential Revision: [D23221733](https://our.internmc.facebook.com/intern/diff/D23221733) [ghstack-poisoned]
Summary: Porting op tests from test_quantize_jit.py Test Plan: TestQuantizeFxOps Reviewers: Subscribers: Tasks: Tags: Differential Revision: [D23221733](https://our.internmc.facebook.com/intern/diff/D23221733) [ghstack-poisoned]
Summary: Porting op tests from test_quantize_jit.py Test Plan: TestQuantizeFxOps Reviewers: Subscribers: Tasks: Tags: Differential Revision: [D23221733](https://our.internmc.facebook.com/intern/diff/D23221733) [ghstack-poisoned]
Summary: Porting op tests from test_quantize_jit.py Test Plan: TestQuantizeFxOps Reviewers: Subscribers: Tasks: Tags: Differential Revision: [D23221733](https://our.internmc.facebook.com/intern/diff/D23221733) [ghstack-poisoned]
Summary: Porting op tests from test_quantize_jit.py Test Plan: TestQuantizeFxOps Reviewers: Subscribers: Tasks: Tags: Differential Revision: [D23221733](https://our.internmc.facebook.com/intern/diff/D23221733) [ghstack-poisoned]
raise Exception("unhandled conv type:", type(self.conv)) | ||
quantized = qconv.from_float(self.conv) | ||
# TODO: make the mapping configurable? | ||
assert type(self.conv) in DEFAULT_MODULE_MAPPING, \ |
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.
This is a looser check than the previous check which was checking for specific module classes. Is this sufficient?
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.
yeah it is equivalent, the check on the left side can be represented by the mapping.
This pull request has been merged in 9984d33. |
Stack from ghstack:
Summary:
Porting op tests from test_quantize_jit.py
Test Plan:
TestQuantizeFxOps
Reviewers:
Subscribers:
Tasks:
Tags:
Differential Revision: D23221733