bc贷

搜索 猫眼电影 融媒体矩阵
  • 山东手机报

  • 猫眼电影

  • 大众网官方微信

  • 大众网官方微博

  • 抖音

  • 人民号

  • 全国党媒平台

  • 央视频

  • 百家号

  • 快手

  • 头条号

  • 哔哩哔哩

首页 >新闻 >社会新闻

AMD GPU性能暴涨7倍,优化算法首次开源!高效MoE支持任意专家数量

2025-04-06 08:24:30
来源:

猫眼电影

作者:

吕孟阳

手机查看

  猫眼电影记者 艾中营 报道y85ouztigy6ysf5y

新智元报道

编辑:LRST

【新智元导读】通过完全启用并发多块执行,支持任意专家数量(MAX_EXPERT_NUMBER==256),并积极利用共享内存(5kB LDS)和寄存器(52 VGPRs,48 SGPRs),MoE Align & Sort逻辑被精心设计,实现了显著的性能提升:A100提升3倍,H200提升3倍,MI100提升10倍,MI300X/MI300A提升7倍...

MoE(Mixture of Experts)模型模仿了人脑的低功耗运作模式:功能被划分为多个独立的部分,在思考时通过自适应路由部分激活,从而提高计算效率。

牛津大学研究论文中的人脑皮层示意图,来源于互联网

首个可在CUDA真正可行的版本是Switch Transformer[1],随后通过循环利用(Up Cycling)稠密模型Mistral[2]进一步优化了该设计。

SwitchTransformer-MoE

随后,DeepSeek V2/V3/R1[3][4][5]通过引入共享专家[3]和门控偏差(gating bias)[4][5]进一步改进了MoE,最终实现了无辅助损失(auxiliary loss free)的MoE模型 [4][5]。这一优化本质上归因于一个关键事实:当使用共享专家(DeepSeek团队选择的值为1)时,可以通过在较大的专家池(256个上施加偏差分数的惩罚,从而缓解专家路由的不均衡问题[11]。

MoE层本质上是由多个专家前馈网络(FFN)组成的层,其中包含门控函数(gating functions),用于根据Top-K门控分数(DeepSeek V3/R1中引入偏差)进行激活路由,并在所选的FFN层上通过Group GEMM计算logits。

该功能在很大程度上依赖于基数排序(radix sort)逻辑。借助MoE Align & Sort,机器学习研究人员和实践者可以按照专家ID对tokens进行排序。

在某些应用中,例如TransformerEngine[6][7],该操作最初是通过已废弃的cub::DeviceRadixSort实现的,而增加的permute操作用于记录源(左)到目标(右)的映射,其梯度操作为unpermute。

MoE Permute示例

尽管cub::DeviceRadixSort大量使用共享内存,相比于基于__shfl_xor_sync(仅使用线程本地内存)的实现略慢,但它不支持对齐排序(alignment sorting)。

对齐排序对于Group GEMM的效率至关重要,因为它允许专家以块(block 为单位处理tokens。

SGLang 中的MoE Align & Sort算法采用了对齐排序,但在支持多达256个专家的大规模prefill操作时效率并不理想。该问题已在issue#2732中被确认。

目前的实现将MoE Align & Sort拆分为两个kernel启动(kernel launches):

对齐(alignment):在单个block内执行传统基数排序算法对齐后的偏移计算(alignment-based offsets computation);

??时事1:小樱裸体被鸣人❌

??04月06日,独家调查河南大学大礼堂失火事件 最新回应→,

  “金狼部落、罗浮大泽、还有雷族、紫山家,你们方才不是高高在上、完全无视我们吗?怎么现在脸色发白了?”二猛直愣愣的说道。

,男明星裸体洗澡🔞互摸。

??04月06日,文化中国行丨守住千年文脉 他们用更年轻的方式“激活”姑苏城,

  人们接着向下读,当念道“击碎初始地通道”这一行字时,顿时炸锅了。

,Video纲手❌自慰爽naruto,大司命被扒开腿做高H电影,老妇XXXXX性开放中文对白。

??时事2:女初高中打屁股❤ⅹ网站

??04月06日,今年前9个月泰国出口额持续增长,

  ⒏公司科技信息部要作好优质服务工作,各单位也要主动积极与公司科技信息部保持联系,做到“及时沟通、上下同欲”,发挥科技、信息队伍的整体优势。我也想提醒大家一下,总工程师一定要有大局观念,一把手安排的事要干好,副职的急事也要办好,同时要准确定位,一定要种好自己的地,千万别种了人家的田,荒了自己的地。

,▌娜娜▌羞辱榨精挑战在线观看,少年巨茎征服熟女人妻麻妃图片,晚上看秒湿黄文。

??04月06日,东西问|加拿大剧作家施崇梵:为何讲述《推销员在中国》的故事?,

  三方人马对峙,而后毫无意外的起了冲突,林地中出现喊杀声。

,ZoZσZσZσ女人极品另类,lisa裸体图奶头走光图,装睡被陌生人揉到高潮。

??时事3:性护士裸体人体片

??04月06日,腕饰巧搭新国风 年轻人“戴”出新消费,

  黄色雾霭涌动,大地抖动,一头庞然大物长足有五十米,状若穿山甲,但是却生有蛟龙头,威严无比,流淌金色霞光,出现在此地。

,❌❌❌自慰调教av大师网站,福瑞涩涩同人18禁网站免费,张鹤伦左手把好把右手把大梁。

??04月06日,【行藏西 看阿里】河北援藏:十年守护只为阿里妇幼健康,

  “好久远呀。”小石昊惊讶。

,女学生喷浆❌❌❌,91丝袜❤️国产在线观看,被摄政王C哭高肉双腿打开H。

??时事4:美女露出🐻让男生揉的小说亲嘴

??04月06日,李潮东:外企如何探寻与新质生产力相适应的绿色发展之路,

  两个孩子大眼都很亮,相互对视,笑了起来。

,鸣人XX❌XX佐良娜,扒开灰原哀粉嫩小泬喷潮调教,校园恋足 (01-43)。

??04月06日,霍海丹:长征精神是新时代鼓舞中国人民推进中国式现代化的重要法宝,

城镇化工作会议讲话材料 篇14

,美女被c🔞🔞💃💃❌❌,无码熟妇人妻av久久久一区二区,扒开老师❌狂揉❌电视剧。

责编:曹传佳

审核:宁康园

责编:徐孟加

相关推荐 换一换

Copyright (C) 2001- Dzwww   鲁ICP备09023866号-1

Sitemap