tests/kernels/test_top_k_per_row.py · 0bfa229bf1f6b12f215d045f4acb4b9607937f32 · OpenDAS / vllm_cscc · GitLab

Find file Blame History Permalink

[Perf][Kernel] Add faster topKperRow decode kernel for DeepSeek-V3.2 sparse attention (#33680) · afdce12c

Roberto L. Castro authored Feb 10, 2026

Signed-off-by: LopezCastroRoberto <rocastro@redhat.com>
Signed-off-by: Roberto L. Castro <38211239+LopezCastroRoberto@users.noreply.github.com>
Co-authored-by: Claude Sonnet 4.5 <noreply@anthropic.com>

afdce12c

test_top_k_per_row.py 12.1 KB