DeepSeek MLA(多头潜在注意力)深度解析:如何用低秩压缩将 KV Cache 压缩到原来的 1/10
在大模型推理的工程实践中,KV Cache 是一把双刃剑:它通过缓存历史 Token 的 Key 和 Value 向量避免了重复计算,是自回归推理加速的基石;但它同时也是显存消耗的”头号杀手”。以一个 70B 参数模...
在大模型推理的工程实践中,KV Cache 是一把双刃剑:它通过缓存历史 Token 的 Key 和 Value 向量避免了重复计算,是自回归推理加速的基石;但它同时也是显存消耗的”头号杀手”。以一个 70B 参数模...

在大模型落地应用的过程中,一个愈发普遍的需求是:同一个基座模型(如 LLaMA-3-70B),需要同时服务数千个不同的下游任务。每个任务有自己的微调权重——客服对话、代码补全、文档摘要、多语言翻译……如果为每个任务都部署一个独立模型副本,显...

为什么 Prefix Caching 成为大模型推理的必备优化? 在大模型推理服务的生产环境中,一个经常被忽视但影响巨大的现象是:大量请求共享相同的前缀。无论是系统提示词(System Prompt)、Few-shot 示例,还是多轮对话中...

引言:为什么梯度累积是大模型训练的”隐形基础设施” 在 ChatGPT 横空出世的三年后,大模型训练的”显学”似乎已经从单卡微调转移到了千卡预训练。各类技术文章铺天盖地讨论着 3D 并行、Ze...

引言:为什么大模型训练需要 FP8? 随着大语言模型的参数量从数十亿迈向数千亿,训练过程中的显存消耗和计算吞吐成为核心瓶颈。以 Llama 3 405B 为例,使用 BF16 精度训练仅模型参数就需要约 810GB 显存,加上梯度、优化器状...

引言:Prefill 阶段的”独木桥”困境 在大模型推理的工程实践中,我们常常遇到这样一个矛盾场景:当请求队列中有一个长文本(比如 32K tokens 的法律文档)正在进行 Prefill 处理时,后面排队等候的短...
引言:为什么 Batching 策略决定了推理服务的天花板 在大模型推理服务的工程实践中,有一个经常被低估却至关重要的设计决策:如何将并发的用户请求组织成批次(Batch)送入 GPU 执行。这个看似简单的调度问题,实际上直接决定了服务的吞...
在大型语言模型(LLM)的推理过程中,通常分为两个关键阶段:Prefill(预填充/处理Prompt)阶段和Decode(解码/自回归生成)阶段。这两个阶段对硬件资源的需求截然不同,理解它们的瓶颈对于优化推理性能至关重要。 1. 概念定义:...
在脱离了熟悉的 CUDA 生态后,针对华为昇腾(Ascend)硬件进行深度学习模型推理性能优化,是许多开发者需要面临的挑战。昇腾平台的核心是 CANN(Compute Architecture for Neural Networks)工具链...

从单体到分体:为什么大模型推理正在走向 Prefill/Decode 分离部署? 2024 年下半年以来,大模型推理架构迎来了一个新的趋势——PD 分离部署(Prefill/Decode Disaggregation)。简单来说,就是将传统...