Optimizing GPU Scheduling: Accelerating Megakernel Compilation with CPU Cost Model

yacineMTB · x · 2026-08-04

The post discusses how a compiler explores various ways to schedule work on GPU Streaming Multiprocessors (SMs). Since actually compiling the megakernel is slow, the team built a CPU-only cost model to simulate execution time for a given schedule. This allows them to quickly identify the top candidates, which are then fully compiled to find the optimal scheduling strategy, greatly speeding up the optimization process.

Original post →

More from Infra

Infra channel →