mlsanityData to 14 Oct 2026

FusionML: Prefill, Not Decode - Mechanism and Boundaries of CPU+GPU Co-Execution on Unified-Memory Apple Silicon

arXiv 2607.22785 ↗First posted 24 Jul 2026cs.AR cs.DC cs.LGcited by 1

Not on the map. Why?

Abstract (arXiv)

Apple-Silicon SoCs share CPU, GPU, and Neural Engine over one unified memory system, raising the question of whether transformer inference can be accelerated by splitting single operators across units. Prior attempts, including our own, failed or produced precision-confounded wins. We identify the cause: MLX's lazy-graph scheduler serializes cross-stream work whenever a CPU-stream operation consumes an unmaterialized GPU result inside one evaluation graph, so a row-split matmul that runs x1.38 faster with materialized inputs runs x0.66 slower than GPU-only inside a lazy graph; an eager materialization boundary restores concurrency (x1.34). sys implements a per-layer, contention-aware CPU+GPU row split for transformer prefill built on this fix. Evaluated across five chips and three Apple-Silicon generations, community-replicated, the split accelerates Llama-shaped decoder-block prefill by x1.15--x1.38, unchanged at full 32-block depth, and reaches x1.18--x1.25 faster time-to-first-token on a real Qwen2.5-7B checkpoint served through stock MLX-LM, with token-identical outputs and unchanged decode throughput. We characterize the boundaries equally carefully: decode cannot benefit, bound by shared bandwidth co-execution does not add; precision-matched training loses x0.86--x0.97 on all five chips; ANE dispatch overhead excludes it at layer granularity; and a no-regression runtime gate becomes self-defeating under memory pressure, where probing an alternative mode evicts the active mode's working set. Code, raw results, and generation transcripts are released.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 5 of its 7 references that are in the map. First posted on arXiv 24 Jul 2026.

Builds on

5 of 7 references.

References · 5

  1. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
  2. 2022Fast Inference from Transformers via Speculative Decodingcited by 1,653
  3. 2023Accelerating Large Language Model Decoding with Speculative Samplingcited by 1,006
  4. 2023Splitwise: Efficient generative LLM inference using phase splittingcited by 558
  5. 2023SARATHI: Efficient LLM Inference by Piggybacking Decodes with Chunked Prefillscited by 193

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.