-
Notifications
You must be signed in to change notification settings - Fork 89
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
refactoring quantization passes #2544
Conversation
@@ -41,7 +41,7 @@ struct module; | |||
*/ | |||
struct MIGRAPHX_EXPORT capture_arguments_pass | |||
{ | |||
std::vector<std::string> ins_names = {"dot", "convolution"}; | |||
std::set<std::string> ins_names = {"dot", "convolution"}; |
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.
Do you think unordered_set
would be better?
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.
yes for sure. I can make change
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.
done
Codecov ReportAttention:
❗ Your organization needs to install the Codecov GitHub app to enable full functionality. Additional details and impacted files@@ Coverage Diff @@
## develop #2544 +/- ##
========================================
Coverage 91.41% 91.42%
========================================
Files 452 452
Lines 17153 17148 -5
========================================
- Hits 15681 15678 -3
+ Misses 1472 1470 -2 ☔ View full report in Codecov by Sentry. |
quantize_8bits_pass
wasn't really usingins_names
field. Removed it.using
std::set
for keeping names of the operators to be quantized instead ofstd::vector
for easy lookup.