setup.py · a66cf40b205d57ac1b5dc96b6bb6f8e813b18316 · OpenDAS / vllm_cscc · GitLab

"vscode:/vscode.git/clone" did not exist on "71470bc4afdab89eccc232b668a69571ffede1dc"

Find file Blame History Permalink

[Kernel][ROCm][AMD] enable fused topk_softmax kernel for moe layer (#4927) · a66cf40b
Divakar Verma authored Jun 02, 2024
```
This PR enables the fused topk_softmax kernel used in moe layer for HIP
```
a66cf40b

setup.py 14.1 KB