跳到主要內容
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

要點與分析由自動化流程生成,可能有誤,請結合原始來源核實。