技术栈

TensorRT-LLM中的MoE并行推理

smartcat20102025-01-09 12:31

2种并行方式:

moe_tp_size:按照维度切分,每个GPU拥有所有Expert的一部分权重。

moe_ep_size: 按照Expert切分,每个GPU有用一部分Expert的所有权重。

二者可以搭配一起使用。

限制:二者的乘积,必须等于模型并行(不是指moe_tp_size,而是整个模型的tp_size)的卡数。

上一篇:【leetcode100】二叉树的直径
下一篇:HTML课堂之搜索工具箱/讲师duluo
相关推荐
Eloudy
2 天前
全文 - CDS: Deadline-Aware, Multi-Chiplet GPU Scheduler in Gem5
gpu·chiplet
Eloudy
5 天前
全文 - 第 03 章 - Principles and Practices of Interconnection Networks
gpu
Eloudy
5 天前
全文 - 第 01 章 - Principles and Practices of Interconnection Networks
gpu·chiplet
Eloudy
5 天前
全文 - 第 08 章 - Principles and Practices of Interconnection Networks
gpu
Eloudy
5 天前
全文 - 第 02 章 - Principles and Practices of Interconnection Networks
gpu
Eloudy
6 天前
全文 - 第 06 章 - Principles and Practices of Interconnection Networks
gpu
Eloudy
6 天前
BookSim (1.0) 用户手册
gpu
论文复现现场
7 天前
AutoDL、算家云与公有云 GPU 怎么选?环境复现、计费与断点恢复对比
pytorch·深度学习·云计算·gpu
Eloudy
8 天前
全文 - Dissecting How Die Scaling Breaks GPU Fine-grained Scheduling
gpu·chiplet
Eloudy
9 天前
BlackWell 双 计算 die 的 L2 一致性 cases
gpu·chiplet
热门推荐
01如何新建文件夹? 电脑新建文件夹的4种方法02国内可直接用、免费额度/永久免费的大模型API清单(含 SiliconFlow、火山、阿里、智谱、百度、Kimi、DeepSeek、DMXAPI 等)03历年考研数学一、数学二、数学三真题试卷及答案PDF04GitHub 镜像站点05轻量化AI渗透武器库|ARTEX自主攻防系统、资产梳理、漏洞挖掘、链路分析全流程智能化落地06蜘蛛磁力 搜索引擎大全,如何使用蜘蛛磁力查找磁力链接07Windows 10/11 如何退出微软账户并改用本地账户登录(含找不到选项的解决方法)08Telegram APK 下载安装完整指南 — 2026年最新09微信历史版本含下载地址( Windows PC | 安卓 | MAC )及设置微信不更新10「全语录+省流版」邵艾伦 × 孙宇晨:聊风口、聊个人IP、聊年轻人如何抓住 AI 时代的机会