6f1904aa8c1929dbf398055a43eff384fcb67229
Probe data (probe_moe_fused_breakdown.sh on BI-V100): F.linear loop 8 experts: 8.060 ms bmm pre-transposed full MoE: 6.918 ms ← 14% faster transpose+contiguous runtime: 22.219 ms ← why CUTLASS was 27ms Changes: - Lazy-cache w13_t (E,H,2I) and w2_t (E,I,H) on first decode call - FC1: torch.bmm(x_expand, w13_t_sel) replaces F.linear(x, w13_sel.reshape) - FC2: torch.bmm(act, w2_t_sel) replaces torch.bmm(w2_sel, act^T) - Zero runtime transpose cost after first call
project_6
Description
Languages
C++
41.8%
Cuda
31.6%
Python
22.2%
C
2.1%
CMake
1.1%
Other
1.1%