Repository navigation
[Quantization] add humming moe backend to all dense/moe oracles - #41652
Conversation
Signed-off-by: Jinzhen Lin <jinzhen.ljz@antgroup.com>
There was a problem hiding this comment.
Code Review
This pull request integrates the Humming MoE backend across various quantization schemes, including FP8, INT8, WNA16, MXFP4, and NVFP4. Key changes include updating the backend oracles to support Humming, implementing weight conversion logic for Humming kernels, and refactoring quantization methods to pass the layer instance to kernel and configuration factories. Additionally, the code now correctly references local_num_experts instead of num_experts in Humming-specific modules. I have no feedback to provide.
Signed-off-by: Jinzhen Lin <jinzhen.ljz@antgroup.com>
|
Hi @jinzhen-lin, the pre-commit checks have failed. Please run: uv pip install pre-commit>=4.5.1
pre-commit install
pre-commit run --all-filesThen, commit the changes and push to your branch. For future commits, Tip Is
|
Signed-off-by: Jinzhen Lin <jinzhen.ljz@antgroup.com>
|
This pull request has merge conflicts that must be resolved before it can be |
Signed-off-by: Jinzhen Lin <jinzhen.ljz@antgroup.com>
Signed-off-by: Jinzhen Lin <jinzhen.ljz@antgroup.com>
Signed-off-by: Jinzhen Lin <jinzhen.ljz@antgroup.com>
Signed-off-by: Jinzhen Lin <jinzhen.ljz@antgroup.com>
|
This pull request has merge conflicts that must be resolved before it can be |
Signed-off-by: Jinzhen Lin <jinzhen.ljz@antgroup.com>
Signed-off-by: Jinzhen Lin <jinzhen.ljz@antgroup.com>
Signed-off-by: Jinzhen Lin <jinzhen.ljz@antgroup.com>
Signed-off-by: Jinzhen Lin <jinzhen.ljz@antgroup.com>
Signed-off-by: Jinzhen Lin <jinzhen.ljz@antgroup.com>
Signed-off-by: Jinzhen Lin <jinzhen.ljz@antgroup.com>
Signed-off-by: Jinzhen Lin <jinzhen.ljz@antgroup.com>
…-project#41652) Signed-off-by: Jinzhen Lin <jinzhen.ljz@antgroup.com> Co-authored-by: mgoin <mgoin64@gmail.com>
…-project#41652) Signed-off-by: Jinzhen Lin <jinzhen.ljz@antgroup.com> Co-authored-by: mgoin <mgoin64@gmail.com>
…-project#41652) Signed-off-by: Jinzhen Lin <jinzhen.ljz@antgroup.com> Co-authored-by: mgoin <mgoin64@gmail.com>
This PR add humming moe backend to all dense/moe oracles, and fixed several minor bugs discovered during model testing.
Tested with
RedHatAI/Qwen3.5-4B-quantized.w4a16nm-testing/Qwen2.5-VL-3B-Instruct-quantized.w8a8nm-testing/Qwen2-1.5B-Instruct-FP8W8nm-testing/Llama-3.1-8B-Instruct-KV-FP8-Tensornm-testing/Qwen3-0.6B-FP8_BLOCKnvidia/Qwen3-8B-NVFP4mgoin/Qwen3-0.6B-MXFP8Qwen/Qwen3-4B-FP8Qwen/Qwen2.5-3B-Instruct-GPTQ-Int4Qwen/Qwen1.5-32B-Chat-AWQRedHatAI/Qwen3.6-35B-A3B-NVFP4nm-testing/Qwen3-30B-A3B-Fp8-v1nm-testing/Qwen3-30B-A3B-FP8-blockRedHatAI/Qwen3-30B-A3B-Instruct-2507-quantized.w8a8nvidia/Qwen3-30B-A3B-NVFP4Qwen/Qwen3.5-35B-A3B-FP8openai/gpt-oss-20bQwen/Qwen3.5-35B-A3B-GPTQ-Int4QuantTrio/Qwen3.6-35B-A3B-AWQQwen/Qwen3.5-35B-A3Bwith--quantization experts_int8NOTE:
VLLM_HUMMING_INPUT_QUANT_CONFIG. In the future, it will default to using the activation format from the checkpoint's config.json.--moe-backend hummingfor moe kernel and add--linear-backend hummingfor linear kernel