vllm/tests/quantization
Michael Goin 978aed5300
[Kernel][Attention] Separate `Attention.kv_scale` into `k_scale` and `v_scale` (#6081)
2024-07-16 15:31:32 -07:00
..
__init__.py [CI/Build] Move `test_utils.py` to `tests/utils.py` (#4425) 2024-05-13 23:50:09 +09:00
test_bitsandbytes.py [CI/Build][REDO] Add is_quant_method_supported to control quantization test configurations (#5466) 2024-06-13 15:18:08 +00:00
test_compressed_tensors.py [ Misc ] Refactor Marlin Python Utilities (#6082) 2024-07-11 15:40:11 +00:00
test_configs.py [mypy] Enable type checking for test directory (#5017) 2024-06-15 04:45:31 +00:00
test_fp8.py [Kernel][Attention] Separate `Attention.kv_scale` into `k_scale` and `v_scale` (#6081) 2024-07-16 15:31:32 -07:00
test_lm_head.py [CORE] Quantized lm-head Framework (#4442) 2024-07-02 22:25:17 +00:00
utils.py [hardware][misc] introduce platform abstraction (#6080) 2024-07-02 20:12:22 -07:00