tests/kernels/test_attention.py · 4934d492744d14104353b8236ef8a0405edf1622 · SIYIXNI / vllm · GitLab

"ISSUE_TEMPLATE" did not exist on "94635139b1060a008ef0e5fb1faf22ad60bec7b9"

Find file Blame History Permalink

Replace head_mapping params with num_kv_heads to attention kernel. (#1997) · dacaf5a4
wbn authored Dec 11, 2023
```
Co-authored-by: wangguoya <wangguoya@baidu.com>
Co-authored-by: Yang Zhao <zhaoyangstar@foxmail.com>
```
dacaf5a4

test_attention.py 10.6 KB