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
"csrc/opt/attention_kernels_opt.cu" did not exist on "e41f06702cb6d6787ac4474832264108a6e28780"
46450b8d33eeee3c619e4d6aea6652ee3d16386f
Switch branch/tag
vllm_cscc
tests
v1
tpu
test_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
test_sampler.py
3.41 KB
Edit
Web IDE
Replace test_sampler.py
×
Attach a file by drag & drop or
click to upload
Commit message
Replace test_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>