跳到正文
NVIDIA Technical Blog· Yang Liu·· 3 小时前AI 评分31

在 NVIDIA DGX Spark 上加速 MoE 推理的 3 种方法

3 Ways to Boost MoE Inference on NVIDIA DGX Spark

AI 导读

NVIDIA 技术博客给出在 DGX Spark 上加速 MoE 推理的 3 种方法,基于对 Inclusion AI 的 Ling-3.0-flash 在 batch-size-1 下的性能分析。profiling 显示解码阶段受内存带宽限制,读模型权重主导执行时间,MoE 计算与输出层投影是两大瓶颈。

正文 · 原文

Abstract green computing cube with layered panels, performance icons, and streams of data on a black backgroundGenerating tokens faster starts with identifying what limits inference performance. In the batch-size-1 evaluation of Ling-3.0-flash from Inclusion AI on NVIDIA...Abstract green computing cube with layered panels, performance icons, and streams of data on a black background

Generating tokens faster starts with identifying what limits inference performance. In the batch-size-1 evaluation of Ling-3.0-flash from Inclusion AI on NVIDIA DGX Spark, profiling showed that decoding was memory-bound: reading model weights drove execution time. Two operations stood out—the mixture-of-experts (MoE) computation and the output-layer projection that produces scores across the model’…

Source

来源:NVIDIA Technical Blog · developer.nvidia.com