mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-10-31 08:13:15 +08:00
tilelang/quantize's identity is device-side conversion helpers plus host-side data-prep utilities (quantization.py's _tir_* converters, the lop3/mxfp intrin groups); a complete quantizer operator does not belong in the library. Move the TileLang BF16->NVFP4 kernel factory and its dispatcher to maint/gemm/tilelang_nvfp4_quantizer.py, next to the maint benchmark that consumes it. nvfp4.py now carries exactly the format contract: the _tl_* device helpers (FP4/UE4M3 encode/decode, scale-source coordinates) any user kernel can call, the blockscaled_chunk_kmajor packers/swizzles, and the torch reference quantizer — 524 lines (1334 at series start). The package re-export list shrinks to the four user-facing names; the alias wrappers are gone. Validated: quantize + example CLI suites (35 passed), maint benchmark and example --verify pass end-to-end. Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>