[Perf][Spec Decode] Fused vocab-parallel argmax for LogitsProcessor.get_top_tokens() - #60099

[Perf][Spec Decode] Fused vocab-parallel argmax for LogitsProcessor.get_top_tokens()#60099

pst2154 wants to merge 1 commit into

vllm-project:mainvllm-project/vllm:mainfrom

pst2154:vocab-parallel-argmax-kernelspst2154/vllm:vocab-parallel-argmax-kernelsCopy head branch name to clipboard