TensorRT-LLM中的MoE并行推理

2种并行方式:

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

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

二者可以搭配一起使用。

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

相关推荐
Smoothcloud润云13 小时前
GPU算力平台推荐:复现实验时,如何判断租用资源是否合适?
服务器·人工智能·深度学习·gpu算力·gpu
Eloudy2 天前
从源码 安装sglang
gpu
Eloudy2 天前
SGLang 的组成结构与基本工作原理
llm·gpu·sglang
Eloudy3 天前
`ibv_get_device_list` 实现分析
网络·gpu·rdma
蒸鱼Yuzheng3 天前
移动端 Overdraw 治理:从单帧热图到屏幕空间预算
性能优化·gpu·renderdoc·overdraw·移动端渲染
Eloudy3 天前
MLNX_OFED 源码补齐、Debug/Release 构建命令与修改记录
gpu·rdma
Smoothcloud润云4 天前
远程纳管GPU怎么安全升级?驱动、组件灰度与回滚实战
人工智能·深度学习·安全·gpu算力·gpu·数据库管理员
鲨鱼辣钊4 天前
KVM显卡直通实战_RTX4090分时复用
gpu·kvm·vfio
Eloudy4 天前
从 huggingFace 下载大模型命令备忘
llm·gpu·rdma
Smoothcloud润云4 天前
GPU远程纳管安全吗?最小权限、网络边界与日志审计实战
服务器·网络·人工智能·安全·gpu算力·gpu