-
Notifications
You must be signed in to change notification settings - Fork 169
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
Refactor the API for quant method argument for quantize function #400
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/400
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit 20a2938 with merge base 512b5d6 (): This comment was automatically generated by Dr. CI and updates every 15 minutes. |
test/integration/test_integration.py
Outdated
int8da_int8w, | ||
int4_weight_only, | ||
int8_weight_only, | ||
int8_dynamic_act_int8_weight, |
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.
can we expand out act into activation as well? Might as well since we hardly have any abbreviations left
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.
cool!
can you also include the update to the main README as well in this PR - specifically the snippet in https://github.com/pytorch/ao?tab=readme-ov-file#without-intrusive-code-changes |
28680ff
to
e27b898
Compare
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.
yea i think these function names are more clear and definitely better than the string api
5fb966d
to
91a6dc5
Compare
Summary: Addressing feedback from pytorch#384 and pytorch#375 Test Plan: regression tests python test/quantization/test_quant_api.py python test/integration/test_integration.py Reviewers: Subscribers: Tasks: Tags:
Summary: * added repr for TensorCoreTiledAQTLayoutTensor: pytorch#397 * removed the str -> apply_tensor_class map that was used to implement string APIs, which is removed in pytorch#400 Test Plan: CI Reviewers: Subscribers: Tasks: Tags:
Summary: * added repr for TensorCoreTiledAQTLayoutTensor: pytorch#397 * removed the str -> apply_tensor_class map that was used to implement string APIs, which is removed in pytorch#400 Test Plan: CI Reviewers: Subscribers: Tasks: Tags:
…orch#400) Summary: Addressing feedback from pytorch#384 and pytorch#375 Test Plan: regression tests python test/quantization/test_quant_api.py python test/integration/test_integration.py Reviewers: Subscribers: Tasks: Tags:
Summary: * added repr for TensorCoreTiledAQTLayoutTensor: pytorch#397 * removed the str -> apply_tensor_class map that was used to implement string APIs, which is removed in pytorch#400 Test Plan: CI Reviewers: Subscribers: Tasks: Tags:
Summary:
Addressing feedback from #384 and #375
Test Plan:
regression tests
python test/quantization/test_quant_api.py
python test/integration/test_integration.py
Reviewers:
Subscribers:
Tasks:
Tags: