tests/kernels/test_top_k_per_row.py · e1098ced95146d98a4ed46c81ee709013d54fb1f · OpenDAS / vllm_cscc · GitLab

Find file Blame History Permalink

Add topk logits torch op for DS3.2. (#25945) · e1098ced

Daniel Cámpora authored Oct 07, 2025

Signed-off-by: Daniel Campora <961215+dcampora@users.noreply.github.com>
Signed-off-by: Daniel Cámpora <961215+dcampora@users.noreply.github.com>
Co-authored-by: youkaichao <youkaichao@gmail.com>

e1098ced

test_top_k_per_row.py 4.5 KB