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

引言:为什么大模型训练需要 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)。简单来说,就是将传统...
在资源受限的端侧设备(如手机、IoT设备)上部署深度学习模型时,模型量化(通常是转换为INT8)是降低延迟和功耗的关键技术。然而,量化方案并非只有一种。本文将对比静态离线量化(Static Post-Training Quantizatio...
在现代深度学习分布式训练中,NVIDIA Collective Communications Library (NCCL) 是实现高性能 GPU 间通信的核心工具。NCCL 提供了多种通信算法来优化 All-Reduce、Broadcast...

一、引言:当模型参数放不进单卡显存时 随着大语言模型规模的不断膨胀,从 BERT-base 的 1.1 亿参数到 Llama 3 的 4050 亿参数,单张 GPU 的显存早已无法承载完整的模型训练。即便是拥有 80GB HBM3 显存的 ...
随着大模型的体积不断增长,如何在资源受限的端侧设备上高效运行这些模型成为了关键挑战。Apple M系列芯片,特别是最新的M3系列,通过其独特的统一内存架构(Unified Memory Architecture, UMA),为端侧大模型推理...