ft支持w4a8, w4a8 moe, fp8 量化(仅针对SM90 卡子) - #706
Conversation
|
ENV INFO: OS: Ubuntu 22.04.5 LTS (x86_64) Python version: 3.12.3 | packaged by Anaconda, Inc. | (main, May 6 2024, 19:46:43) [GCC 11.2.0] (64-bit runtime) CPU: Versions of relevant libraries: ==============================
|
|
验证的模型: |
|
w4a8 算子功能,性能的测试 CPU Instruction Info: [AVX2: ON] [AVX512F: ON] [AVX512_VNNI: ON] [AVX512_BF16: ON] [AMX: ON] == op: linear == Summary: PASS w4a8 forward_check测试 2026-07-30 22:48:37,907 - INFO - Namespace(model='/root/autodl-tmp/Meta-Llama-3-8B-Instruct-W4AFP8-AWQ', tokens=8, hf_device='cuda', flm_dtype='bfloat16', flm_atype='bfloat16', flm_threads=4, flm_device='cuda') Compressing model: 0%| | 0/224 [00:00<?, ?it/s] Loading weights: 0%| | 0/739 [00:00<?, ?it/s] 0%| | 0/1 [00:00<?, ?it/s][transformers] Ignoring clean_up_tokenization_spaces=True for BPE tokenizer TokenizersBackend. The clean_up_tokenization post-processing step is designed for WordPiece tokenizers and is destructive for BPE (it strips spaces before punctuation). Set clean_up_tokenization_spaces=False to suppress this warning, or set clean_up_tokenization_spaces_for_bpe_even_though_it_will_corrupt_output=True to force cleanup anyway. 100%|██████████| 1/1 [00:07<00:00, 7.42s/it] 0%| | 0/1 [00:00<?, ?it/s] Decompressing model: 0%| | 0/224 [00:00<?, ?it/s]�[A Decompressing model: 15%|█▌ | 34/224 [00:00<00:00, 333.32it/s]�[A Decompressing model: 43%|████▎ | 97/224 [00:00<00:00, 505.04it/s]�[A Decompressing model: 71%|███████▏ | 160/224 [00:00<00:00, 561.36it/s]�[A Decompressing model: 100%|█████████▉| 223/224 [00:00<00:00, 588.21it/s]�[A 100%|██████████| 1/1 [00:02<00:00, 2.02s/it] |
w4a8 prefill性能测试[1/3] start_case: w4a8-prefill-128
|
fp量化算子功能 性能测试CPU Instruction Info: [AVX2: ON] [AVX512F: ON] [AVX512_VNNI: ON] [AVX512_BF16: ON] [AMX: ON] == op: w4a8_activation_quant == Summary: PASS |
w4a8 moe 算子功能 性能测试CPU Instruction Info: [AVX2: ON] [AVX512F: ON] [AVX512_VNNI: ON] [AVX512_BF16: ON] [AMX: ON] == op: mergemoe_w4a8 == Summary: PASS |
ft支持 w4a8, w4a8 moe 算子, fp8 量化算子