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

在 NVIDIA DGX Spark 上提升 MoE 推理性能的 3 种方法

3 Ways to Boost MoE Inference on NVIDIA DGX Spark

AI 导读

在 NVIDIA DGX Spark 上对 Inclusion AI 的 Ling-3.0-flash 做 batch-size-1 评测时,性能分析显示解码受限于内存带宽,读取模型权重主导了执行时间。其中 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 Developer Blog · developer.nvidia.com