python/sglang/srt/layers/moe/cutlass_moe.py · 3e56f557fdd107c678e3ea563c01baed0f6ef80b · zhaoyu6 / 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

cutlass_moe.py 14 KB