Heterogeneous Computing for AI Agent Inference

Authors: Aaron Zhao, Junyi Liu | Affiliations: Imperial College London, Microsoft Research | PDF: Heterogeneous_Computing_AI_Agent_Inference_2026.pdf

一句话总结

本文用 Operational Intensity (OI)Capacity Footprint (CF) 刻画 AI agent 推理:agent 工作流(coding/WUA/CUA)上下文雪崩至 300K–1M tokens,暴露 roofline 未覆盖的 memory capacity wall,论证未来需 compute/network/memory 系统级异构

核心贡献

  1. OI + CF 双指标:补全 roofline/MFU/MBU 对 memory capacity 限制的盲区(“双低”区域)
  2. Agent 工作负载画像:Chatbot vs Coding vs Web-use vs Computer-use 在 LLaMA-70B 上 CF/OI 差异巨大
  3. Snowballing context:Agent 多轮交互(~20–30 次/任务)使 CF 快速超过单卡 B200 HBM
  4. 架构/优化映射:MHA/GQA/MLA、MoE、量化、prefill-decode 分离各改变 OI/CF 象限
  5. 异构 scaling 论点:加卡解 CF 但不提 OI — 需跨 compute、互联、memory 的 cohesive datacenter 设计

关键数字

设置结果
Coding agent context300K–1M tokens (typical snowball)
Env interactions / task20–30 (reported)
CF vs B200Most agent CF exceeds single-card capacity at modest batch
Decode OIExtremely low — DRAM load dominates over compute

与 wiki 交叉引用

Citations

[1] Heterogeneous_Computing_AI_Agent_Inference_2026.pdf — Zhao & Liu (2026) [2] heterogeneous-computing-ai-agent-inference.md — 结构化摘录