biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity
Files
8f1729b829795c2c98152fef8857c5e7b8c4e648
vllm/csrc/moe/topk_softmax_kernels.cu
Divakar Verma a66cf40b20 [Kernel][ROCm][AMD] enable fused topk_softmax kernel for moe layer (#4927) 
This PR enables the fused topk_softmax kernel used in moe layer for HIP
2024-06-02 14:13:26 -07:00

20 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink