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