跳到主要内容
AI News HubLIVE
更多
仅摘要

mKernel:多GPU、多节点融合内核库,实现GPU驱动通信

加州大学伯克利分校UCCL团队发布mKernel,将节点内NVLink、节点间RDMA和密集计算融合到单个持久CUDA内核中,旨在减少AI工作负载中的通信开销。研究显示通信可占用前向传播43.6%和训练总时间32%的时间。mKernel提供五种融合内核,支持ConnectX-7和AWS EFA后端。

来源MarkTechPost作者: Asif Razzaq
mKernel:多GPU、多节点融合内核库,实现GPU驱动通信
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明

本文仅提供摘要

根据来源内容权限,本站仅提供编辑摘要、要点和原文链接。

展开要点与分析

文章情报

工程师进阶

要点

  • mKernel将节点内NVLink、节点间RDMA和计算融合到单个持久CUDA内核中
  • 通信开销在MoE模型中最高可占执行时间的47%
  • 提供五种融合内核:AllGather+GEMM、GEMM+AllReduce、MoE Dispatch+GEMM、Ring Attention和GEMM+ReduceScatter
  • 基于libibverbs实现GPU发起的RDMA,不依赖NCCL或NVSHMEM

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。