bc贷

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

  • 猫眼电影

  • 大众网官方微信

  • 大众网官方微博

  • 抖音

  • 人民号

  • 全国党媒平台

  • 央视频

  • 百家号

  • 快手

  • 头条号

  • 哔哩哔哩

首页 >新闻 >社会新闻

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

2025-04-03 21:16:27
来源:

猫眼电影

作者:

郑爽

手机查看

  猫眼电影记者 郑永和 报道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月03日,国家外汇管理局:2023年银行累计结汇155336亿元人民币,

  这是一场浩劫,席卷了这片大地,在吞天雀还有那只顶天立地的凶兽的带领下,山河被染红了,所有族群全灭。

,警察Sm俱乐部男警奴被调教。

??04月03日,江西新干上演箱包T台秀 当地民众演绎时尚新风潮,

  不远处,狈村狩猎队伍中的头领狈山冷笑,正是他一箭射穿了小不点的手臂,他脸色苍白,不久前曾被银月剖开肚腹。他再次弯弓,躲在灌木丛中,开始了新一轮的袭杀,不理会青鳞鹰,只瞄准小不点的心脏或咽喉。

,霍雨浩对着萧萧下面疯狂进攻,女仆舌吻❌吸乳舌吻羞羞,动漫美女裸体被涩涩动画软件。

??时事2:A🔞W看🔴䏒🔞❌❌二次元

??04月03日,【央视快评】加快发展新质生产力 扎实推进高质量发展,

  他因此而力量暴涨,整具躯体由两米竟不可思议的拔高到了三米,强壮了一大截,浑身绕着一条条闪电般的光束,血气惊人。

,丝袜美女裸体被❌出水在线观看,小入视频动漫,凪光老师为何困倒。

??04月03日,第46个植树节回头看:中国人种树到底有多牛?,

  半数以上的人都已猜到小不点的身份,知道他有多么的凶残,因此在那些师兄师姐还没有回来的情况下就去一战,依旧是无惧。

,XXNX96👙HD,r星⭐精选,japanの排泄脱粪排泄。

??时事3:老师脱👙让男学生摸🐻

??04月03日,(经济观察)中国首发支持银发经济专门文件 回应三大需求,

  为了集中精力抓好九年级毕业班教育教学工作,切实提高教育教学质量,全面提高学生素质,使每一个毕业班学生能顺利地完成学业,切实提高学区和学校在全县中考成绩中的位次,学校要高度重视毕业班工作。开校以来学校做好了以下工作:

,趁她洗澡摸她胸把她c了她,jmcomic.2.0官网下载,甘雨被❌到爽🔞高潮痉挛小说。

??04月03日,广州郑仙诞文化旅游周启动 “千年古诞”展新姿,

  四是规划管理较软。主要表现在城市规划力度不够,乡(镇)、村规划欠缺,规划科学性不足,严肃性和权威性不够。我市农村“小洋楼”普及率超过,有些地方达到,但是缺乏规划,大多是单门独户的随意建设,与豫北相比反差很大。同时,规划实施的保障机制不健全,“规划规划,纸上画画,墙上挂挂”;或者“今天规划,明天不算”,随意修改。规划的质量也有待提高,缺乏高起点、高标准、高品位,缺乏精细度。

,原神内衣,男男GaYGAYS✅18体育生,岳伦我轻点太深了。

??时事4:麻豆久久久久久久久丝袜

??04月03日,万千气象丨海底“植树造林”这件事 我们真的是太卷了,

  “这是一头恶灵,我会想办法将它除掉!”小不点语气坚定,再三向族人保证,绝不会逞一时之勇,如果不对头,他会立刻逃遁。

,小樱裸体羞羞喷水wwwAPP,表演绝技bb吸烟的是谁,嫩草❤️❤️影院。

??04月03日,澳网公开赛正赛首日,白卓璇王欣瑜出局,

  本届邻居节,将以“邻居节的五日,即芳邻日,互助日,健康日,娱乐日,  日”为中心,以“深化和谐社区建设、促进邻里友爱团结、喜迎奥运、弘扬爱国主义精神”为目标,以健康、文明、积极向上的系列活动,展示我社区三个文明建设的丰硕成果和全社区邻里和睦、居民友爱、安居乐业的勃勃生机。

,18禁坤坤浏览器下载,二型糖尿病能恢复正常吗,福瑞e621的网站。

责编:吴厚琴

审核:祝二辉

责编:韩宝生

相关推荐 换一换

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

Sitemap