NVIDIA 开发者博客· Yang Liu·· 2 小时前精选AI 评分52
NVIDIA DGX Spark 上提升 MoE 推理性能的三种方法
3 Ways to Boost MoE Inference on NVIDIA DGX Spark
AI 导读
NVIDIA 开发者博客介绍在 DGX Spark 上提升 MoE 推理性能的三种方法。对 Inclusion AI 的 Ling-3.0-flash 做 batch-size-1 评测时,profiling 显示解码处于 memory-bound 状态,读取模型权重主导执行时间,其中 MoE 计算和产生全模型打分的输出层投影两项操作最为突出。
推荐理由
原文以 Ling-3.0-flash 实测指出 batch-size-1 解码受显存带宽限制,为优化 MoE 推理提供了排查思路。
来源:NVIDIA 开发者博客 · developer.nvidia.com