sgl-kernel/python/sgl_kernel/moe.py · 3e56f557fdd107c678e3ea563c01baed0f6ef80b · change / sglang · GitLab

Find file Blame History Permalink

Add a CUDA kernel for fusing mapping and weighted sum for MoE. (#6916) · 3e56f557
Elfie Guo authored Jun 07, 2025
```
Co-authored-by: Elfie Guo <elfiegxf@gmail.com>
```
3e56f557

moe.py 5.67 KB