biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity
Files
ec784b2526219cd96159a52074ab8cd4e684410a
vllm/setup.py
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

14 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink