Skip to content

FEP(sig-framework): FlagOS-Compressor model quantization framework - #91

Open
wbavon wants to merge 7 commits into
mainfrom
fep-flagos-compressor
Open

FEP(sig-framework): FlagOS-Compressor model quantization framework#91
wbavon wants to merge 7 commits into
mainfrom
fep-flagos-compressor

Conversation

@wbavon

@wbavon wbavon commented Jul 30, 2026

Copy link
Copy Markdown
Collaborator

FEP for the first release (v0.1.0) of FlagOS-Compressor, the FlagOS weight-format conversion and quantization tool, in the FlagOS 2.2 cycle.

Scope is pinned to the release branch release/v0.1.0 (cut at eecf2fa, tagged v0.1.0-rc1), not main:

  • BF16 dequantization of MXFP4 (E2M1 + E8M0) and block FP8 (128x128 tile + E8M0).
  • Weight-only MSE INT4/INT8 → compressed-tensors W4A16 and W8A16 (pack-quantized), plus dynamic per-token W8A8 (int-quantized).
  • Module selection (linear, attention, mlp, moe, moe.routed, moe.shared), regex select/exclude, and versioned YAML recipes.
  • inspect, --dry-run, validate; CPU and CUDA backends.
  • Structural coverage: dense, 2D routed/shared experts, MLA attention naming, and fused 3D routed-expert banks for the one registered layout (Qwen3.5-MoE).

Notable scope corrections made by checking the release branch rather than main:

New module and repository, so this FEP doubles as the module admission proposal.

Status is Implementable: implementation is complete on the release branch (PRs #1#5 merged, v0.1.0-rc1 tagged 2026-08-24) and the Test Plan carries executable commands with expected results. Owner: @rdzhu225. Acceptance runs during the 2.2 testing window flip it to Implemented. Open TODOs carried in doc comments, to be settled during acceptance: numerical tolerance for the dequant/quant checks, host-memory sizing guidance, dependency pins / wheel publication, and the serving stack + datasets for the load tests. Feature Freeze for 2.2 is 2026-08-31.

@CLAassistant

CLAassistant commented Jul 30, 2026

Copy link
Copy Markdown

CLA assistant check
All committers have signed the CLA.

@wbavon wbavon added FEP FlagOS Enhancement Proposal sig/framework SIG: framework target/2.2 Target release: FlagOS 2.2 labels Aug 1, 2026
wbavon added 3 commits August 2, 2026 11:52
Verified every claim against the release/v0.1.0 branch (eecf2fa) rather
than main:

- INT8, per-channel INT8, and dynamic-token W8A8 (PRs #3/#5) are in the
  release branch, so they move from Non-Goals into G2 with their actual
  storage formats and defaults.
- Calibrated GPTQ/AWQ/AutoRound (PR #6) landed on main after the branch
  was cut; recorded as out of scope for 2.2.
- Backends are cpu and cuda only; other chips are covered by generating
  on a CPU/CUDA host and validating the load on target.
- G4 states the fused-3D-expert limit: only the Qwen3.5-MoE layout is
  registered, other axis orders are refused.
- Test Plan now carries the exact commands and expected results the
  template requires, including the guard/negative cases.
- Packaging answers the wheel question; PRs #4/#5 added to Related PRs.
- Drop the (Required)/(Optional) template markers; no merged FEP keeps
  them.
- Remove the FlagCompressor rename note, which has no bearing on the
  proposal.
- Cut editorializing ("deliberate", "conservative baseline", "not to be
  bypassed", "machine-readable", "correct behavior") and redundant
  clauses, keeping the technical claims unchanged.
@wbavon
wbavon force-pushed the fep-flagos-compressor branch from 9b727e6 to 0e8d8e3 Compare August 25, 2026 06:11
@wbavon

wbavon commented Aug 25, 2026

Copy link
Copy Markdown
Collaborator Author

recheck

Implementation is complete on release/v0.1.0 (PRs #1-#5, v0.1.0-rc1
tagged at eecf2fa on 2026-08-24) and the Test Plan carries executable
commands, so the design-phase Provisional status no longer reflects
reality. Acceptance is still ahead; Implemented waits on the test
matrix.
@wbavon
wbavon requested a review from Al2O3Al2O3 August 25, 2026 06:22
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

FEP FlagOS Enhancement Proposal sig/framework SIG: framework target/2.2 Target release: FlagOS 2.2

Projects

Status: Todo

Development

Successfully merging this pull request may close these issues.

2 participants