tests/kernels/moe/test_moe.py · fc7980dbdae7a2eb91b680a15f2e3e44059e3ecd · OpenDAS / vllm_cscc · GitLab

Find file Blame History Permalink

[torch.compile] Speed up MOE handling in forward_context (#33184) · 611b1875
Richard Zou authored Jan 27, 2026
```
Signed-off-by: Richard Zou <zou3519@gmail.com>
(cherry picked from commit d9aa39a3)
```
611b1875

test_moe.py 48 KB