让大模型推理在崩溃后能快速恢复,无需重启或改代码。
Concordia: JIT-Compiled Persistent-Kernel Checkpointing for Fault-Tolerant LLM Inference

- 用设备级持久内核实现故障时自动保存状态,不依赖主机干预。
- 支持对注意力、适配器等关键组件的增量式检查点,恢复速度快。
- 适合需要长时间运行的大模型服务,如客服机器人、科研助手。
长时运行的大模型代理在GPU上保留了大量关键状态:键值缓存、请求调度器、通信状态,甚至在线适配器。一旦出现GPU或通信器故障,这些状态丢失将导致数分钟至数小时的工作付诸东流。现有恢复机制要么重启整个服务栈,要么要求每个注意力层和运行时组件都嵌入特定检查点逻辑。本文提出Concordia,一种基于设备端持久内核的容错推理运行时。它通过介入GPU模块加载,在PTX与SASS层级实现代码插桩,可在框架与库边界以下插入检查点和暂停钩子。针对每个注册的大模型状态区域,Concordia即时编译专用的差分检查点处理程序(如KV块扫描器、适配器页扫描器或恢复应用器),并热替换至持久内核的操作符表中。该持久内核使用无锁环形缓冲区接收计算、检查点、日志追加和恢复任务,同一执行器持续检测脏页、生成差分数据,并将已提交记录写入CXL内存或主机DRAM中的可见日志,实现高效可靠的状态恢复。
原文摘要 · Abstract (English)
Long-running LLM agents keep valuable state resident on GPUs: KV caches, request schedulers, communication state, and sometimes online adapters. Losing this state after a GPU or communicator failure can discard minutes to hours of work, yet existing recovery mechanisms either restart the whole serving stack or require application-specific checkpoint logic inside every attention and runtime component. This paper argues that fault tolerance for such workloads needs a GPU-resident execution context: checkpoint hooks must run at device synchronization points, observe binary kernels that frameworks and libraries actually execute, and recover without putting the host CPU on the critical path. We present Concordia, a runtime that uses a device-resident persistent kernel as the substrate for fault-tolerant LLM inference. Concordia interposes on GPU module loading and supports PTX- and SASS-level instrumentation, allowing checkpoint and pause hooks to be inserted below framework code and library boundaries. For each registered LLM state region, Concordia JIT-compiles a specialized delta-checkpoint handler -- for example, a KV-block scanner, adapter-page scanner, or recovery applier -- and hot-swaps it into the persistent kernel's operator table. The persistent kernel consumes a lock-free ring buffer of compute, checkpoint, append-log, and recovery tasks, so the same always-on executor triggers dirty-page detection, stages deltas, and appends committed records to a CPU-visible log in CXL memory or host DRAM.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。