Skip to content
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

Add static quant to float8 #787

Merged
merged 6 commits into from
Sep 5, 2024

Conversation

jainapurva
Copy link
Contributor

@jainapurva jainapurva commented Sep 1, 2024

This PR adds support for Float8 static quant to affine quantized tensor.

Added a sample recipe for float8 static quantization for calibration flow.

Copy link

pytorch-bot bot commented Sep 1, 2024

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/787

Note: Links to docs will display an error until the docs builds have been completed.

✅ No Failures

As of commit 8b9b6d9 with merge base ba2d3b1 (image):
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Sep 1, 2024
@jainapurva jainapurva marked this pull request as ready for review September 5, 2024 17:41
Copy link
Contributor

@jerryzh168 jerryzh168 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thanks!

@jainapurva jainapurva merged commit a246d87 into main Sep 5, 2024
17 checks passed
andrewor14 pushed a commit that referenced this pull request Sep 5, 2024
* Add static quant

* Merged static_quant intx - floatx

* Merged static_quant intx - floatx

* Add assert for mapping type

* Add assert for mapping type

* Update intx_static to support floatx call
HDCharles pushed a commit that referenced this pull request Sep 9, 2024
* Add static quant

* Merged static_quant intx - floatx

* Merged static_quant intx - floatx

* Add assert for mapping type

* Add assert for mapping type

* Update intx_static to support floatx call
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants