vLLM
/
vllm
mirror of https://github.com/vllm-project/vllm.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
vllm/csrc/moe
History
Woosuk Kwon f0d4e14557
Add fused top-K softmax kernel for MoE (#2769) 		2024-02-05 17:38:02 -08:00
..
moe_ops.cpp Add fused top-K softmax kernel for MoE (#2769) 2024-02-05 17:38:02 -08:00
moe_ops.h Add fused top-K softmax kernel for MoE (#2769) 2024-02-05 17:38:02 -08:00
topk_softmax_kernels.cu Add fused top-K softmax kernel for MoE (#2769) 2024-02-05 17:38:02 -08:00