AIToday

RaMP framework optimizes Mixture-of-Experts inference by dynamically selecting kernel configurations based on runtime expert routing, achieving 1.22x kernel speedup and 1.30x end-to-end speedup in vLLM serving

arXiv cs.LGApr 30, 20261 min read

Summaries like this, in your inbox every morning.

Sign up free →

3 Key Points

  1. RaMP is a routing-aware dispatch framework that selects the fastest kernel configuration from a runtime expert histogram, achieving 0.93% mean regret versus exhaustive search using a four-parameter wave cost model fitted from 10-24 minutes of one-time profiling per model.

  2. A performance-region analysis derived from hardware constants alone predicts when each optimization helps, correctly predicting all 8 tested architectures including 3 unseen ones. Paired with a CuTe DSL kernel exposing 134-268 polymorphic configurations, RaMP delivers 1.22x kernel speedup over static dispatch.

  3. In vLLM serving, RaMP achieves 1.30x end-to-end speedup over Triton, 1.41x over DeepGEMM, and 1.13x over FlashInfer CUTLASS. When applied to Alpha-MoE, it delivers 1.14x speedup with no source modification, since the model depends only on CTA grid geometry and is kernel-agnostic.

Discussion

No discussion yet for this article

Stay ahead with AI news

Get curated AI news from 200+ sources delivered daily to your inbox. Free to use.

Get Started Free

Free · takes 30 seconds · unsubscribe anytime

1 minute a day. The AI essentials.

200+ sources · Email / LINE / Slack

Get it free →