Skip to content

Fix float8 + int4 QAT #2851

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

Merged
merged 1 commit into from
Aug 22, 2025
Merged

Fix float8 + int4 QAT #2851

merged 1 commit into from
Aug 22, 2025

Conversation

andrewor14
Copy link
Contributor

Summary: After #2779, Float8DynamicActivationInt4Weight no longer has the group_size field, but QAT continues to read from this field. This commit fixes it to just use the fixed 128 group size.

Test Plan:

python test/quantization/test_qat.py -k test_infer_fp8_int4_config

**Summary:** After #2779,
`Float8DynamicActivationInt4Weight` no longer has the `group_size`
field, but QAT continues to read from this field. This commit
fixes it to just use the fixed 128 group size.

**Test Plan:**
```
python test/quantization/test_qat.py -k test_infer_fp8_int4_config
```
Copy link

pytorch-bot bot commented Aug 22, 2025

🔗 Helpful Links

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

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

❌ 1 Cancelled Job, 1 Unrelated Failure

As of commit 4e9f17d with merge base e7251df (image):

CANCELLED JOB - The following job was cancelled. Please retry:

BROKEN TRUNK - The following job failed but were present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

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

@andrewor14 andrewor14 requested a review from jerryzh168 August 22, 2025 18:00
@meta-cla meta-cla 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 Aug 22, 2025
@andrewor14 andrewor14 added the topic: bug fix Use this tag for PRs that fix bugs label Aug 22, 2025
@andrewor14 andrewor14 merged commit 0596713 into main Aug 22, 2025
16 of 20 checks passed
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. topic: bug fix Use this tag for PRs that fix bugs
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants