Dockerfile.rocm · 5d7e3d0176e0dbcf144c64b7d14d996c55e36c50 · OpenDAS / vllm_cscc · GitLab

"vllm/vscode:/vscode.git/clone" did not exist on "0d4044edd85de30d7d4558aeea4d1e95c7c556d6"

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

Dockerfile.rocm 3.77 KB