TA的每日心情 | 开心
2020-4-8 10:45 |
|---|
签到天数: 227 天 [LV.7]分神
|
/ s7 ^4 k2 J& b1 @% |# `在论文里，这是第3.2.2节的内容
h) {& G( \& X* \1 l0 u" Z. `* V
3.2.2. Efficient Implementation of Cross-Node All-to-All Communication
& l, o, C0 Q" V8 S% v9 g1 z) P4 {9 Y" vIn order to ensure sufficient computational performance for DualPipe, we customize efficient5 N h" \( N4 ]4 V8 r
cross-node all-to-all communication kernels (including dispatching and combining) to conserve
) W* P) H' T4 B8 F& ]the number of SMs dedicated to communication. The implementation of the kernels is codesigned with the MoE gating algorithm and the network topology of our cluster. To be specific,
* F" z* l/ g) c) Q6 @! c1 win our cluster, cross-node GPUs are fully interconnected with IB, and intra-node communications
/ c1 G+ b6 z/ f5 o" Iare handled via NVLink. NVLink offers a bandwidth of 160 GB/s, roughly 3.2 times that of IB0 n5 X5 c8 a8 h' B6 ^ R' Y
(50 GB/s). To effectively leverage the different bandwidths of IB and NVLink, we limit each$ t1 f7 ^+ \8 ?* ?, U
token to be dispatched to at most 4 nodes, thereby reducing IB traffic. For each token, when its, e" o2 |& @: a t
routing decision is made, it will first be transmitted via IB to the GPUs with the same in-node
: [) P3 @! H% [index on its target nodes. Once it reaches the target nodes, we will endeavor to ensure that it is
- f0 i) Z8 j! U* M& }$ oinstantaneously forwarded via NVLink to specific GPUs that host their target experts, without
4 a. ?4 k0 }$ n0 d( Y6 V: {! x4 g+ lbeing blocked by subsequently arriving tokens. In this way, communications via IB and NVLink# U9 R$ p/ q* p+ u- S
are fully overlapped, and each token can efficiently select an average of 3.2 experts per node4 F$ w( P- i9 c/ p) T. r! m! E
without incurring additional overhead from NVLink. This implies that, although DeepSeek-V3
" m' F* a: M8 K Y13
9 \* T; g8 {( X5 I: Q" O; \. L5 mselects only 8 routed experts in practice, it can scale up this number to a maximum of 13 experts
9 U4 s" C6 H. }) u: Q(4 nodes × 3.2 experts/node) while preserving the same communication cost. Overall, under
, ?1 \4 G- F! W- k) usuch a communication strategy, only 20 SMs are sufficient to fully utilize the bandwidths of IB* N) P& e( _5 _7 l) J. ?9 Y
and NVLink.
/ }, s$ `, K6 M& w" }In detail, we employ the warp specialization technique (Bauer et al., 2014) and partition7 Y9 W+ r+ u J/ R, r
20 SMs into 10 communication channels. During the dispatching process, (1) IB sending, (2) d. t$ n& J0 f" U5 t
IB-to-NVLink forwarding, and (3) NVLink receiving are handled by respective warps. The
: x4 Y+ ^8 o* C" I2 n& xnumber of warps allocated to each communication task is dynamically adjusted according to the
- ^5 e* U! H b, c4 ~actual workload across all SMs. Similarly, during the combining process, (1) NVLink sending,: v; a8 d- H; i8 u# s8 F
(2) NVLink-to-IB forwarding and accumulation, and (3) IB receiving and accumulation are also
% c! R0 d2 ~9 chandled by dynamically adjusted warps. In addition, both dispatching and combining kernels$ \5 D* G* V p9 y9 d' Y- X+ H
overlap with the computation stream, so we also consider their impact on other SM computation3 R9 f, V4 k7 O8 j! ?* v
kernels. Specifically, we employ customized PTX (Parallel Thread Execution) instructions and8 c" T* ]) D/ J& S! Z
auto-tune the communication chunk size, which significantly reduces the use of the L2 cache
2 ^% J9 l$ I& f* \and the interference to other SMs.
* B' l" p, i3 n; }3 P3 H; s/ J& @# K- _% G$ f$ H$ W+ q* f8 t0 l
通俗一点说，就是为了实现高效的跨节点全面通信。解决的问题本质上和唐家山老师日志里说的双机对拷的场景差不多。一般来说单机多卡之间用nvlink，多机多卡之间依赖IB网络，但nvlink的速率是IB网络的速率的3.2倍，需要通过一些优化来实现更好的传输策略。这是一整套方案。
, v" h. _3 V. V$ q7 p
) D3 Z; A8 t" ^, A我的理解，使用PTX在其中，是为了更精准的定制线程执行减少通信块分配传输之间的串扰。
. c5 W6 K5 ]: ?& ]! n, [! N+ T" S! \2 @
目的不是为了绕cuda，反而是为了让cuda的效率更高。
+ I5 S& O( Y2 Y8 C& }
' l4 r& x$ b' G类比一下，就好比发现网卡驱动在对拷特定内存块的时候会和应用的线程执行出现串行导致效率降低，而绕开操作系统定义的与网卡驱动的接口，直接使用网卡支持的指令集进行了优化。 |
评分
-
查看全部评分
|
雷达卡
楼主
收藏
分享
淘帖
支持
反对
提升卡
置顶卡
沉默卡
喧嚣卡
变色卡
抢沙发
千斤顶
显身卡
发表于 2025-2-4 01:15:04