TaxBreak: Unmasking the Hidden Costs of LLM Inference Through Overhead Decomposition
Organizations: NeuroAI Computer Architecture Lab (NCAL), Electrical and Computer Engineering Department, Carnegie Mellon University
Abstract
Large Language Model (LLM) inference is widely used in interactive assistants and agentic systems. In latency-sensitive deployments, inference time can become dominated by host-side overheads. Existing approaches typically expose this cost only as an aggregate residual or a launch/queue metric, which is often insufficient to identify which execution layer should be optimized. This work presents TaxBreak, a trace-driven methodology for decomposing host-visible orchestration overhead into three components: framework translation time, CUDA library translation time, and kernel launch-path time. We validate TaxBreak on NVIDIA H100 and H200 systems and use it to derive our proposed Host-Device Balance Index (HDBI), a boundedness summary index that relates device-active execution to host-visible orchestration. Across representative dense and mixture-of-experts workloads in both prefill and decode, we show that aggregate latency, GPU inactivity, or boundedness ratios alone can obscure the dominant optimization target. TaxBreak instead distinguishes cases where optimization should reduce software-stack overhead from cases where the primary win comes from reducing device-side work. We further show that MoE models dispatch 8-11x more kernels per output token than dense models, and that for such host-bound workloads, CPU single-thread performance is a first-order parameter: a faster host CPU reduces orchestration overhead by 10-29% and improves end-to-end latency by up to 14%, even when paired with a slower-clocked GPU. These results position TaxBreak as a diagnostic tool for assessing whether optimization effort should target the software stack or the device-side workload execution.
Figures & tables
| Work | Tax Granularity | CPU-GPU | Cross-Layer Overhead | Workloads | Prefill + Decode | Hopper-Era HW | Evaluation Platform(s) |
| AI Tax [ 25 ] | Pipeline-level | ✗ | ✗ | Google FaceNet, R-CNN | ✗ | ✗ | Edge data center deployment |
| Framework Tax [ 14 ] | Coarse residual | ✗ | ✗ | BERT-Base; ResNet-50; DistilBERT; GPT-2; WavLM. | ✗ | ✗ | NVIDIA Pascal, Turing, Ampere. |
| TKLQT / Kernel-Tax View [ 30 ] | Launch-path only | ✓ | ✗ | Bert-Base-Uncased; XLM-Roberta-Base; GPT-2; Llama-3.2-1B. | ✗ | ✓ | PCIe A100/H100 and tightly-coupled GH200 (prefill-centric). |
| GPU Inference Characterization [ 31 ] | Device-centric | ✗ | ✗ | Llama-3-8B; Qwen2.5-7B/-32B; Qwen3-30B-A3B. | ✓ | ✗ | NVIDIA A100 80GB (SXM) + Jetson AGX Orin. |
| This Work (TaxBreak) | Host-stack attribution: , , | ✓ | ✓ | Llama-3.2-1B,-3B, OLMoE-1B/7B; Qwen1.5-MoE-A2.7B | ✓ | ✓ | H100 (DGX H100); H200 NVL (single-GPU). |
| (a) Library-mediated | (b) Framework-native |
| torch_op | torch_op |
| (nn.Linear.call) | (layer_norm) |
| aten::addmm | aten::native_layer_norm |
| cuBLAS front-end |
| Metric | Llama-3.2-1B | Llama-3.2-3B | OLMoE-1B/7B | Qwen1.5-MoE-A2.7B |
| Total kernel launches | 8,475 | 15,369 | 93,053 | 66,951 |
| Unique kernel names | 77 | 76 | 222 | 223 |
| Kernels per token | 847.5 | 1,536.9 | 9,305.3 | 6,695.1 |
| Diversity ratio | 0.0091 | 0.0049 | 0.0024 | 0.0033 |
| GPU utilization (%) | 58.9 | 67.6 | 15.5 | 27.7 |
| GPU | avg | p50 | p5 | p95 |
| H100 | 4.707 | 4.578 | 4.260 | 5.396 |
| H200 | 4.503 | 4.452 | 4.177 | 4.909 |
| Llama-3.2-3B ( s) | OLMoE-1B/7B ( s) | |||||||
| Kernel Family | ||||||||
| (null) | 4.75 | 4.75 | – | – | 4.73 | 4.73 | – | – |
| Scan (prefix) | 5.07 | 5.16 | 0.32 | 7% | 5.12 | 5.13 | 0.40 | 8% |
| Elem. (unroll) | 5.11 | 5.28 | 0.36 | 8% | 5.11 | 5.20 | 0.38 | 8% |
| Elem. (vector) | 5.13 | 5.38 | 0.38 | 8% | 5.30 | 5.49 | 0.58 | 12% |
| Reduce | 5.30 | 5.30 | 0.55 | 12% | 5.07 | 5.46 | 0.34 | 7% |
LLM Inference Efficiency
Large language model inference efficiency reduces the time, memory, energy, and cost of generating useful outputs.
See all 350 papers