HomeAI & Machine LearningOperator Fusion & the Roofline Model

Operator Fusion & the Roofline Model

Interactive 3D lab: watch a neural-network op pipeline (Conv-BN-ReLU-Conv-Pool) fuse into single kernels, cutting DRAM round-trips, and see the operating point move on a live roofline plot as precision and edge hardware change.

AI & Machine Learning3DAdvanced60 FPS📱 Mobile-adapted⇄ 2D version
ds-topic-52 ↗ Open standalone

Edge-inference runtimes like TensorFlow Lite, Core ML and ONNX Runtime spend as much effort fusing operators as they do quantizing weights — because on battery-powered hardware, DRAM traffic often costs more time and energy than the arithmetic itself. This lab renders a five-stage Conv-BatchNorm-ReLU-Conv-Pool pipeline as a 3D scene: data packets travel up from a memory plane into each operator and back down again, unless that operator has been fused into its neighbor, in which case the packet stays in flight and never touches memory. A live roofline plot — the standard performance-modeling tool for exactly this problem — tracks arithmetic intensity and achieved throughput as you toggle fusion, switch precision (FP32/FP16/INT8) and swap between a mobile CPU, mobile GPU and edge NPU, showing numerically why fusion is worth more on memory-bound stages than compute-bound ones.

⚙ Under the hood

Watch a Conv-BatchNorm-ReLU-Conv-Pool inference pipeline fuse into single kernels, cutting DRAM round-trips, while a live roofline plot tracks arithmetic intensity and achieved throughput across precisions and edge hardware.

edge aimodel compressionoperator fusionroofline modelinference optimizationtinyml

3D · Three.js / WebGL renderer · 60 FPS target · runs fully client-side, no install

What did you find?

Add reproduction steps (optional)