一次长上下文性能翻倍的复盘GLM-5.2 为什么要复用 Indexer 结果作者元宝专栏元宝说 AI 安全记录时间2026-08-222026-08-24 复核120K 上下文的单次请求从 105.694 秒降到 52.130 秒接近 128K 的请求从 118.863 秒降到 57.238 秒。这不是换了更大的机器也不是把某个“加速开关”打开后的偶然结果。第 22 轮真正做的事情很朴素把模型配置里已经写明的层复用关系接到执行代码上。这次优化给我的最大提醒是长上下文服务变慢时未必是硬件不够快也未必需要立刻换量化、换并行策略。有时性能损失只是因为“模型知道自己可以复用”而运行时并没有真正执行复用。先把结果摆在桌上测试对象是 GLM-5.2-NVFP48 张 A100TP8 EP8最大上下文 131,072。每个档位都是单会话、串行执行并发为 1关闭 thinkingtemperature0固定生成 128 个 token每档重复 3 次。档位Prompt tokens优化前 P50 E2E第 22 轮 P50 E2E延迟变化优化后总处理吞吐120K119,751105.694 s52.130 s-50.68%2,299.45 total tok/s近 128K128,859118.863 s57.238 s-51.84%2,253.53 total tok/s优化后的原始样本也比较集中120K52.130337 / 52.139292 / 52.130080s近 128K57.237754 / 57.259347 / 57.234080s两档波动都小于 0.05%。这至少说明结果不是单次“撞上缓存”或偶发的短请求。这里的total tok/s是(prompt_tokens completion_tokens) / 完整请求墙钟时间固定输出只有 128 token所以输出端到端速度仍然约为 2.242.46 output tok/s。提升主要发生在长 Prompt 的预填充阶段而不是解码阶段。慢在哪里78 层都在重复做同一类工作GLM-5.2 的稀疏注意力不是每一层都需要重新计算 indexer 的 Top-K 结果。模型配置里已经给出了这组信息num_hidden_layers78 index_topk_freq4 index_skip_topk_offset3 21 个 full 层 57 个 shared 层full层负责计算新的 Top-Kshared层应该复用此前的选择结果。问题出在运行时的启用条件。固定的fix-38006分支虽然已经有通用 IndexCache 框架但旧逻辑只在配置显式提供use_index_cachetrue时进入。GLM-5.2 的配置没有这个字段于是执行路径退回到默认行为78 层全部运行 indexer。这类问题很容易被误判成“模型太大”或“稀疏注意力本身不适合长上下文”。实际上模型结构和运行时能力并没有完全接上。第 22 轮采用的层映射逻辑可以简化成下面这样# 简化示意展示决策逻辑不是完整补丁freqgetattr(config,index_topk_freq,1)offsetgetattr(config,index_skip_topk_offset,2)patterngetattr(config,index_topk_pattern,None)layer_idextract_layer_index(prefix)ifpatternisnotNoneand0layer_idlen(pattern):skip_topkpattern[layer_id]Selse:skip_topk(max(layer_id-offset1,0)%freq!0)关键变化不是增加一套新算法而是取消对use_index_cache的单一依赖直接使用 GLM 配置中的频率、偏移量和层模式决定是否复用。第 22 轮的变更边界为了避免把多个变量混在一起这一轮只修改了deepseek_v2.py中的 IndexCache 层映射逻辑。没有改动以下内容模型权重和 NVFP4 量化格式BF16 KV CacheTP8、EP8 拓扑128K 上下文上限max-num-batched-tokens8192CUDA、PyTorch、Triton 版本。生产启动脚本仍然使用原来的服务参数优化藏在源码补丁中。这一点很重要如果只复制启动命令、没有同步源码服务可以正常启动却不会得到这轮收益。先过正确性再谈性能第 22 轮没有用“服务能启动”作为成功标准而是设置了三道门8 个 worker 正常启动/health返回 HTTP 200/v1/models返回GLM-5.2-NVFP4最大上下文为 131,072固定 3K/4K 请求共 10 次全部成功结果为 10/10。还有一个需要单独说明的事实原稳定基线在同一组固定短上下文门禁中是 3K0/5、4K0/5而第 22 轮为 5/5、5/5。这说明补丁至少没有引入可见的短上下文错误并且修复了原基线暴露出的固定样例问题。但这不是完整的知识能力评测不能把 10/10 直接写成“模型准确率提升”。资源侧也重新核对过GPU KV Cache 从 136,384 tokens 变为 139,648 tokens单卡 CUDA Graph 显存约从 1.02 GiB 降到 0.88 GiB。服务没有 OOM、NCCL 错误或 Traceback。这组数字能不能直接宣传成“快了一倍”可以说它是强服务端加速证据但不能忽略测试口径的差异。优化前的稳定基线主要来自公网 SSE第 22 轮为了排除公网代理偶发丢失尾帧的问题改为在 worker 内网访问127.0.0.1:8000使用非流式完整 JSON 计时。两边的模型、上下文长度和输出长度一致但网络路径并非完全相同。因此更严谨的表述是在相同模型、硬件、上下文档位和单会话条件下第 22 轮的服务端长上下文墙钟表现约提升 2 倍若要形成严格 A/B 结论还需要让优化前后使用同一个客户端、同一个网络路径重新配对测试。另外不能把日志里的瞬时 Prompt throughput 当成请求级 KPI也不能把 128 个输出 token 除以整次请求时间后再称为“模型解码速度”。这次优化主要减少的是 prefill 等待。从这次排查里沉淀出的复用方法如果以后遇到类似的长上下文性能问题我会按下面的顺序排查先看模型配置有没有描述“可复用性”关注层模式、Top-K 频率、共享层偏移、KV Cache 布局等字段。配置里出现了这些字段不代表运行时已经消费它们。再看执行代码的 guard重点搜use_xxx、硬件判断、模型类型判断和默认值。很多“功能已经支持”的判断最后会被一个没有在模型配置中出现的布尔字段挡住。性能基线要和业务请求同口径长上下文必须记录完整墙钟、Prompt tokens、Completion tokens、P50/P99 和错误率。短 decode 的 tok/s、服务端十秒滚动窗口、单次异常快样本都不能替代端到端数据。每次只改一个变量本轮没有同时调整量化、并行、KV 精度和调度参数因此即使收益很大也能把原因收敛到 IndexCache 层映射而不是把结果归因给一堆“可能有效”的开关。把回滚当成优化的一部分补丁应用前要保存原文件启动脚本要单独留档性能 JSON、准确度报告和日志不能覆盖。若准确度门禁下降、同口径性能回退或出现 OOM恢复备份源码再启动第 21 轮的基础脚本。这次真正值得记住的结论第 22 轮的价值不只是把两个数字变小了。它证明了一件在推理系统里很常见、却经常被忽略的事性能瓶颈可能位于“模型结构声明”和“运行时执行路径”的接缝处。当模型已经告诉运行时哪些层可以共享结果时最划算的优化往往不是继续堆硬件而是先确认这段信息有没有真的穿过配置解析、模块初始化和 kernel 调度。这次方案已经达到当前实验的保留标准但它仍有清晰边界比较结果需要在统一网络路径下继续复测128K 并发没有被验证也不能把短样例门禁当作完整准确度结论。对生产部署来说性能收益、正确性门禁和可回滚性必须同时成立少一个都不能算完成。具体部署的详细步骤在https://blog.csdn.net/weixin_43833637/article/details/163851066?spm1011.2124.3001.6209