tests/kernels/attention/test_attention_selector.py · a87cc508599d2257f39420536dd5bb52aaa557c3 · OpenDAS / vllm_cscc

[Attn,KV-cache] Use per-head scales in the attention selector (#34281) · a87cc508

Eldar Kurtić authored Feb 24, 2026


Signed-off-by: Your Name <you@example.com>
Signed-off-by: Eldar Kurtic <research@neuralmagic.com>
Co-authored-by: Eldar Kurtic <research@neuralmagic.com>
Co-authored-by: Your Name <you@example.com>

a87cc508

test_attention_selector.py 14 KB

Replace test_attention_selector.py