vllm/utils/math_utils.py · c656ba3b4d2cda82ca753eefde4b10cbf04c0a3f · OpenDAS / vllm_cscc

[Kernel] Triton-based Top-k and Top-p sampler kernels (#33538) · c656ba3b

Jongseok Park authored Feb 17, 2026


Signed-off-by: js_park <cakeng@naver.com>
Signed-off-by: Jongseok Park <37990712+cakeng@users.noreply.github.com>
Signed-off-by: Sunga Kim <sunga.kim@berkeley.edu>
Signed-off-by: Nick Hill <nickhill123@gmail.com>
Co-authored-by: Sunga Kim <sunga.kim@berkeley.edu>
Co-authored-by: Nick Hill <nickhill123@gmail.com>

c656ba3b

math_utils.py 890 Bytes

Replace math_utils.py