What problem does it solve?
It answers compute-level questions about LLM serving—operator order, tensor dimensions, per-op FLOPs, and MFU—without requiring a full profiling run, and validates static estimates against real torch profiler traces.
Core Features & Use Cases
- Static compute simulation: Builds the per-layer operator sequence with tensor shapes and FLOPs for indexed models (DeepSeek-V3/V4, Qwen3, Kimi-K2, MiniMax-M2/M3, GLM-5) under configurable batch size, sequence length, TP/DP/EP, GPU, and dtype.
- MFU estimation: Computes overall, per-layer, per-operator, and per-kernel MFU from measured latency, with fp8 peak-FLOPS correction for fp8 kernels and GPU specs for H20, H100, H200, and B200.
- Trace validation: Extracts the real operator flow from torch profiler traces (record_shapes=True) and compares it against the static template, escalating to SGLang/vLLM/TensorRT-LLM source code when traces are insufficient.
- Use Case: Given a decode serving shape for Qwen3-235B-A22B on 8 H20 GPUs with a measured 15ms forward pass, produce the full operator table, total FLOPs, and kernel-level MFU breakdown to identify the dominant compute bottleneck.
Quick Start
Ask the agent to simulate the compute flow and MFU for your model, for example: run a compute simulation for Qwen3-235B-A22B decode with batch size 1, TP 8, EP 8 on H20 GPUs in bf16 with a measured latency of 15 ms.