TA的每日心情 | 开心 2020-4-8 10:45 |
|---|
签到天数: 227 天 [LV.7]分神
|
6 Y+ `# p2 ]1 r$ ]) N在论文里,这是第3.2.2节的内容
3 r' D9 g; i; C$ I$ \7 b
7 R# G" q3 a/ p9 [ F6 t3.2.2. Efficient Implementation of Cross-Node All-to-All Communication/ O; O. o: P ~; W* l* ~: C2 ~" B
In order to ensure sufficient computational performance for DualPipe, we customize efficient2 o/ j$ y6 {0 U4 @/ L4 h9 D
cross-node all-to-all communication kernels (including dispatching and combining) to conserve S0 |) V& \. N# L& T, V
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,/ R1 F% f" y! g) h. y/ {& v5 j% T
in our cluster, cross-node GPUs are fully interconnected with IB, and intra-node communications& i# d3 \4 Z* Q2 _% w. f
are handled via NVLink. NVLink offers a bandwidth of 160 GB/s, roughly 3.2 times that of IB# Y0 q; {5 G7 ~
(50 GB/s). To effectively leverage the different bandwidths of IB and NVLink, we limit each
7 l1 v5 J6 \$ z% }0 d0 o, Y( Mtoken to be dispatched to at most 4 nodes, thereby reducing IB traffic. For each token, when its
7 w0 r, t, o. ]2 H$ p/ Srouting decision is made, it will first be transmitted via IB to the GPUs with the same in-node* v6 V0 @% H* V/ C
index on its target nodes. Once it reaches the target nodes, we will endeavor to ensure that it is
& _) g4 [. Q4 K' G3 {instantaneously forwarded via NVLink to specific GPUs that host their target experts, without3 X$ _9 e: [* E- `( x& S
being blocked by subsequently arriving tokens. In this way, communications via IB and NVLink
" G) d3 u" V# \4 o; `+ {are fully overlapped, and each token can efficiently select an average of 3.2 experts per node1 ]6 B( S) l( k
without incurring additional overhead from NVLink. This implies that, although DeepSeek-V3
- v* d3 l, [, W13. z6 ?$ D6 z7 H& {! O; E o
selects only 8 routed experts in practice, it can scale up this number to a maximum of 13 experts( c# \$ [8 p% |0 z4 n( j: j! Y3 X
(4 nodes × 3.2 experts/node) while preserving the same communication cost. Overall, under
* B: I8 A& ?; a- E% o+ |such a communication strategy, only 20 SMs are sufficient to fully utilize the bandwidths of IB. w$ ?4 ?1 J9 N
and NVLink.0 Y2 N9 m. O+ ~6 z# v- E
In detail, we employ the warp specialization technique (Bauer et al., 2014) and partition# k3 s: f& c0 u9 U/ }6 f. e
20 SMs into 10 communication channels. During the dispatching process, (1) IB sending, (2)8 E0 d6 B! K- V* i! M- H/ s
IB-to-NVLink forwarding, and (3) NVLink receiving are handled by respective warps. The; p# C8 R j R/ W; y
number of warps allocated to each communication task is dynamically adjusted according to the
0 J b! v! F) ~0 A# K7 T0 Eactual workload across all SMs. Similarly, during the combining process, (1) NVLink sending,
6 j+ W0 H# S# g(2) NVLink-to-IB forwarding and accumulation, and (3) IB receiving and accumulation are also/ a, Q1 N, i- `2 w }
handled by dynamically adjusted warps. In addition, both dispatching and combining kernels4 a5 |0 S& o! j) C* u2 `
overlap with the computation stream, so we also consider their impact on other SM computation
; |2 I( R8 R/ ?" O0 Wkernels. Specifically, we employ customized PTX (Parallel Thread Execution) instructions and6 b z3 ], \7 O
auto-tune the communication chunk size, which significantly reduces the use of the L2 cache
" s0 E( S1 z1 Hand the interference to other SMs.3 ^- d$ @9 S# x1 Z0 W
+ e$ }3 u# T! ?: y
通俗一点说,就是为了实现高效的跨节点全面通信。解决的问题本质上和唐家山老师日志里说的双机对拷的场景差不多。一般来说单机多卡之间用nvlink,多机多卡之间依赖IB网络,但nvlink的速率是IB网络的速率的3.2倍,需要通过一些优化来实现更好的传输策略。这是一整套方案。
9 P3 y4 a8 o8 J8 O; q. Z6 e5 i" r7 y2 t& m/ R! _) k
我的理解,使用PTX在其中,是为了更精准的定制线程执行减少通信块分配传输之间的串扰。) D3 W, r) T$ {7 V- k7 b
! D5 a1 Q" N. t6 D/ v目的不是为了绕cuda,反而是为了让cuda的效率更高。
, ?- \+ f% E2 V- s5 N# D
9 t3 F+ b- r) ^类比一下,就好比发现网卡驱动在对拷特定内存块的时候会和应用的线程执行出现串行导致效率降低,而绕开操作系统定义的与网卡驱动的接口,直接使用网卡支持的指令集进行了优化。 |
评分
-
查看全部评分
|