Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Repository
47195057e9f2b93b35579e2ffd26e2e15cd49979
Switch branch/tag
vllm_cscc
vllm
v1
sample
ops
topk_topp_sampler.py
Find file
Blame
History
Permalink
[V1][TPU] Speed up top-k on TPU by using torch.topk (#15242)
· 47195057
Hyesoo Yang
authored
Mar 20, 2025
Signed-off-by:
Hyesoo Yang
<
hyeygit@gmail.com
>
47195057
topk_topp_sampler.py
9.79 KB
Edit
Web IDE
Replace topk_topp_sampler.py
×
Attach a file by drag & drop or
click to upload
Commit message
Replace topk_topp_sampler.py
Replace file
Cancel
A new branch will be created in your fork and a new merge request will be started.
Hyesoo Yang <hyeygit@gmail.com>