Skip to content

Commit 24b4376

Browse files
committed
Revise line 52 to 57
1 parent ed73cac commit 24b4376

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

โ€Žintermediate_source/TP_tutorial.rstโ€Ž

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,11 +50,11 @@
5050
PyTorch์˜ Fully Sharded Data Parallel(FSDP)๋Š” ์ด๋ฏธ ๋ชจ๋ธ ํ•™์Šต์„ ํŠน์ • ์ˆ˜์˜ GPU๋กœ ์กฐ์ •ํ•  ์ˆ˜ ์žˆ๋Š” ๊ธฐ๋Šฅ์„ ๊ฐ–์ถ”๊ณ  ์žˆ์Šต๋‹ˆ๋‹ค. ๊ทธ๋Ÿฌ๋‚˜, ๋ชจ๋ธ ํฌ๊ธฐ์™€ GPU ์–‘ ์ธก๋ฉด์—์„œ ๋ชจ๋ธ ํ•™์Šต์„ ๋” ํ™•์žฅํ•˜๋ ค๋ฉด,
5151
Tensor Parallel๊ณผ FSDP์˜ ๊ฒฐํ•ฉ์ด ํ•„์š”ํ•œ, ๋‹ค์Œ๊ณผ ๊ฐ™์€ ์ถ”๊ฐ€์ ์ธ ๊ณผ์ œ๊ฐ€ ๋‹ค์ˆ˜ ๋ฐœ์ƒํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.
5252

53-
1. ๊ทธ ์„ธ๊ณ„์˜ ํฌ๊ธฐ (GPU ์ˆ˜)๊ฐ€ ์ง€๋‚˜์น˜๊ฒŒ ์ปค์ง์— ๋”ฐ๋ผ (128/256 GPU ์ดˆ๊ณผ), FSDP ์ง‘ํ•ฉ(์˜ˆ๋ฅผ ๋“ค์–ด, ``allgather`` )์€ ring latency๋กœ ์ง€๋ฐฐ๋˜๊ณ  ์žˆ์Šต๋‹ˆ๋‹ค. TP/SP๋ฅผ FSDP ์œ„์— ๊ตฌํ˜„ํ•˜์—ฌ, FSDP๋ฅผ ํ˜ธ์ŠคํŠธ ๊ฐ„์—๋งŒ ์ ์šฉํ•˜์—ฌ FSDP ์„ธ๊ณ„ ํฌ๊ธฐ๋ฅผ 8๊ฐœ๋กœ ์ค„์ผ ์ˆ˜ ์žˆ์œผ๋ฉฐ, ๊ทธ์— ๋”ฐ๋ผ ์ง€์—ฐ ๋น„์šฉ๋„ ๋™์ผํ•˜๊ฒŒ ์ค„์ผ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.
53+
1. GPU ์ˆ˜๊ฐ€ ์ง€๋‚˜์น˜๊ฒŒ ์ปค์ง์— ๋”ฐ๋ผ (128/256 GPU ์ดˆ๊ณผ), FSDP ์ง‘ํ•ฉ(์˜ˆ๋ฅผ ๋“ค์–ด, ``allgather`` )์€ ring latency์— ๋งŽ์€ ์˜ํ–ฅ์„ ๋ฐ›์Šต๋‹ˆ๋‹ค. TP/SP๋ฅผ FSDP ์œ„์— ๊ตฌํ˜„ํ•˜์—ฌ, FSDP๋ฅผ ํ˜ธ์ŠคํŠธ ๊ฐ„์—๋งŒ ์ ์šฉํ•˜์—ฌ FSDP์˜ ๊ทœ๋ชจ๋ฅผ 8๊ฐœ๋กœ ์ค„์ผ ์ˆ˜ ์žˆ์œผ๋ฉฐ, ๊ทธ์— ๋”ฐ๋ผ ์ง€์—ฐ ๋น„์šฉ๋„ ๋™์ผํ•˜๊ฒŒ ์ค„์ผ ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.
5454
2. ์ˆ˜๋ ด ๋ฐ GPU ๋ฉ”๋ชจ๋ฆฌ ์ œํ•œ์œผ๋กœ ์ธํ•ด ๊ธ€๋กœ๋ฒŒ ๋ฐฐ์น˜ ํฌ๊ธฐ๋ฅผ GPU ์ˆ˜๋ณด๋‹ค ๋†’๊ฒŒ ์„ค์ •ํ•  ์ˆ˜ ์—†๋Š” ๋ฐ์ดํ„ฐ ๋ณ‘๋ ฌ ์ฒ˜๋ฆฌ์˜ ํ•œ๊ณ„๋ฅผ ๋‹ฌ์„ฑํ•˜๋ ค๋ฉด, Tensor/Sequence Parallel์ด ๊ธ€๋กœ๋ฒŒ ๋ฐฐ์น˜ ํฌ๊ธฐ๋ฅผ "์ถ”์ •(ballpark)"ํ•˜๊ณ , ๋” ๋งŽ์€ GPU๋กœ ํ™•์žฅํ•˜๋Š” ์œ ์ผํ•œ ๋ฐฉ๋ฒ•์ž…๋‹ˆ๋‹ค.
5555
3. ํŠน์ • ์œ ํ˜•์˜ ๋ชจ๋ธ์—์„œ๋Š” ๋กœ์ปฌ ๋ฐฐ์น˜ ํฌ๊ธฐ๊ฐ€ ์ž‘์•„์ง€๋ฉด, TP/SP๊ฐ€ ๋ถ€๋™ ์†Œ์ˆ˜์  ์—ฐ์‚ฐ(FLOPS)์— ๋” ์ตœ์ ํ™”๋œ ํ–‰๋ ฌ ๊ณฑ ํ˜•ํƒœ๋ฅผ ์ƒ์„ฑํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.
5656

57-
๊ทธ๋ž˜์„œ, ์‚ฌ์ „ ํ•™์Šต์„ ํ•  ๋•Œ, ์ด๋Ÿฌํ•œ ํ•œ๊ณ„๋ฅผ ๋‹ฌ์„ฑํ•˜๋Š” ๊ฐ€์žฅ ์‰ฌ์šด ๋ฐฉ๋ฒ•์€ ๋ฌด์—‡์ผ๊นŒ์š”? ์ง€๊ธˆ๋ถ€ํ„ฐ, ์ˆ˜์‹ญ์–ต ํ˜น์€ ์ˆ˜์กฐ ๊ฐœ์˜ ํ† ํฐ์„ ๊ฐ€์ง„ ๋Œ€๊ทœ๋ชจ ์–ธ์–ด ๋ชจ๋ธ (LLM)์„ ์‚ฌ์ „ ํ•™์Šตํ•˜๋Š” ๊ฒƒ์€ ์ˆ˜์ฒœ์žฅ์˜ GPU๋ฅผ ์‚ฌ์šฉํ•  ๋•Œ์กฐ์ฐจ, ์ˆ˜๊ฐœ์›”์ด ์†Œ์š”๋  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.
57+
์‚ฌ์ „ํ•™์Šต ์‹œ ์ด๋Ÿฌํ•œ ํ•œ๊ณ„๋ฅผ ๊ฒฝํ—˜ํ•˜๋Š” ๊ฒƒ์€ ํ”ํ•œ ์ผ์ž…๋‹ˆ๋‹ค. ํ˜„์žฌ๋กœ์„œ๋Š” ์ˆ˜์‹ญ์–ต ํ˜น์€ ์ˆ˜์กฐ ๋‹จ์œ„์˜ ํ† ํฐ์œผ๋กœ ๋Œ€๊ทœ๋ชจ ์–ธ์–ด ๋ชจ๋ธ(LLM)์„ ํ•™์Šตํ•˜๋ ค๋ฉด ์ˆ˜์ฒœ ๋Œ€์˜ GPU๋ฅผ ์‚ฌ์šฉํ•˜๋”๋ผ๋„ ์ˆ˜๊ฐœ์›”์ด ๊ฑธ๋ฆด ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.
5858

5959
* LLM์„ ๋Œ€๊ทœ๋ชจ๋กœ ํ›ˆ๋ จํ•  ๋•Œ๋Š” ํ•ญ์ƒ ํ•œ๊ณ„ 1์— ๋„๋‹ฌํ•ฉ๋‹ˆ๋‹ค. ์˜ˆ๋ฅผ ๋“ค์–ด, Llama 2 70B ๋ชจ๋ธ์€ 35์ผ ๋™์•ˆ 2์ฒœ๊ฐœ GPU๋กœ ํ›ˆ๋ จ๋˜์—ˆ๊ณ , 2์ฒœ๊ฐœ ๊ทœ๋ชจ์—์„œ๋Š” ๋‹ค์ฐจ์› ๋ณ‘๋ ฌ ์ฒ˜๋ฆฌ๊ฐ€ ํ•„์š”ํ•ฉ๋‹ˆ๋‹ค.
6060
* Transformer ๋ชจ๋ธ์ด ์ปค์ง€๋ฉด (์˜ˆ๋ฅผ ๋“ค๋ฉด, Llama2 70B), ๋น ๋ฅด๊ฒŒ ํ•œ๊ณ„ 2์— ๋„๋‹ฌํ•  ๊ฒƒ์ž…๋‹ˆ๋‹ค. ๋ฉ”๋ชจ๋ฆฌ ๋ฐ ์ˆ˜๋ ด ์ œ์•ฝ ์กฐ๊ฑด ๋•Œ๋ฌธ์— ๋กœ์ปฌ ``batch_size=1`` ์กฐ๊ฑด ์กฐ์ฐจ๋„ FSDP๋ฅผ ๋‹จ๋…์œผ๋กœ ์‚ฌ์šฉํ•  ์ˆ˜ ์—†์Šต๋‹ˆ๋‹ค. ์˜ˆ๋ฅผ ๋“ค์–ด, Llama2 ๊ธ€๋กœ๋ฒŒ ๋ฐฐ์น˜ ํฌ๊ธฐ๋Š” 1K์ด๋ฏ€๋กœ, 2K GPU์—์„œ ์˜ค์ง ๋ฐ์ดํ„ฐ ๋ณ‘๋ ฌ ์ฒ˜๋ฆฌ๋งŒ์œผ๋กœ ์‚ฌ์šฉ๋  ์ˆ˜ ์—†์Šต๋‹ˆ๋‹ค.

0 commit comments

Comments
ย (0)