|
50 | 50 | PyTorch์ Fully Sharded Data Parallel(FSDP)๋ ์ด๋ฏธ ๋ชจ๋ธ ํ์ต์ ํน์ ์์ GPU๋ก ์กฐ์ ํ ์ ์๋ ๊ธฐ๋ฅ์ ๊ฐ์ถ๊ณ ์์ต๋๋ค. ๊ทธ๋ฌ๋, ๋ชจ๋ธ ํฌ๊ธฐ์ GPU ์ ์ธก๋ฉด์์ ๋ชจ๋ธ ํ์ต์ ๋ ํ์ฅํ๋ ค๋ฉด, |
51 | 51 | Tensor Parallel๊ณผ FSDP์ ๊ฒฐํฉ์ด ํ์ํ, ๋ค์๊ณผ ๊ฐ์ ์ถ๊ฐ์ ์ธ ๊ณผ์ ๊ฐ ๋ค์ ๋ฐ์ํ ์ ์์ต๋๋ค. |
52 | 52 |
|
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๊ฐ๋ก ์ค์ผ ์ ์์ผ๋ฉฐ, ๊ทธ์ ๋ฐ๋ผ ์ง์ฐ ๋น์ฉ๋ ๋์ผํ๊ฒ ์ค์ผ ์ ์์ต๋๋ค. |
54 | 54 | 2. ์๋ ด ๋ฐ GPU ๋ฉ๋ชจ๋ฆฌ ์ ํ์ผ๋ก ์ธํด ๊ธ๋ก๋ฒ ๋ฐฐ์น ํฌ๊ธฐ๋ฅผ GPU ์๋ณด๋ค ๋๊ฒ ์ค์ ํ ์ ์๋ ๋ฐ์ดํฐ ๋ณ๋ ฌ ์ฒ๋ฆฌ์ ํ๊ณ๋ฅผ ๋ฌ์ฑํ๋ ค๋ฉด, Tensor/Sequence Parallel์ด ๊ธ๋ก๋ฒ ๋ฐฐ์น ํฌ๊ธฐ๋ฅผ "์ถ์ (ballpark)"ํ๊ณ , ๋ ๋ง์ GPU๋ก ํ์ฅํ๋ ์ ์ผํ ๋ฐฉ๋ฒ์
๋๋ค. |
55 | 55 | 3. ํน์ ์ ํ์ ๋ชจ๋ธ์์๋ ๋ก์ปฌ ๋ฐฐ์น ํฌ๊ธฐ๊ฐ ์์์ง๋ฉด, TP/SP๊ฐ ๋ถ๋ ์์์ ์ฐ์ฐ(FLOPS)์ ๋ ์ต์ ํ๋ ํ๋ ฌ ๊ณฑ ํํ๋ฅผ ์์ฑํ ์ ์์ต๋๋ค. |
56 | 56 |
|
57 | | -๊ทธ๋์, ์ฌ์ ํ์ต์ ํ ๋, ์ด๋ฌํ ํ๊ณ๋ฅผ ๋ฌ์ฑํ๋ ๊ฐ์ฅ ์ฌ์ด ๋ฐฉ๋ฒ์ ๋ฌด์์ผ๊น์? ์ง๊ธ๋ถํฐ, ์์ญ์ต ํน์ ์์กฐ ๊ฐ์ ํ ํฐ์ ๊ฐ์ง ๋๊ท๋ชจ ์ธ์ด ๋ชจ๋ธ (LLM)์ ์ฌ์ ํ์ตํ๋ ๊ฒ์ ์์ฒ์ฅ์ GPU๋ฅผ ์ฌ์ฉํ ๋์กฐ์ฐจ, ์๊ฐ์์ด ์์๋ ์ ์์ต๋๋ค. |
| 57 | +์ฌ์ ํ์ต ์ ์ด๋ฌํ ํ๊ณ๋ฅผ ๊ฒฝํํ๋ ๊ฒ์ ํํ ์ผ์
๋๋ค. ํ์ฌ๋ก์๋ ์์ญ์ต ํน์ ์์กฐ ๋จ์์ ํ ํฐ์ผ๋ก ๋๊ท๋ชจ ์ธ์ด ๋ชจ๋ธ(LLM)์ ํ์ตํ๋ ค๋ฉด ์์ฒ ๋์ GPU๋ฅผ ์ฌ์ฉํ๋๋ผ๋ ์๊ฐ์์ด ๊ฑธ๋ฆด ์ ์์ต๋๋ค. |
58 | 58 |
|
59 | 59 | * LLM์ ๋๊ท๋ชจ๋ก ํ๋ จํ ๋๋ ํญ์ ํ๊ณ 1์ ๋๋ฌํฉ๋๋ค. ์๋ฅผ ๋ค์ด, Llama 2 70B ๋ชจ๋ธ์ 35์ผ ๋์ 2์ฒ๊ฐ GPU๋ก ํ๋ จ๋์๊ณ , 2์ฒ๊ฐ ๊ท๋ชจ์์๋ ๋ค์ฐจ์ ๋ณ๋ ฌ ์ฒ๋ฆฌ๊ฐ ํ์ํฉ๋๋ค. |
60 | 60 | * Transformer ๋ชจ๋ธ์ด ์ปค์ง๋ฉด (์๋ฅผ ๋ค๋ฉด, Llama2 70B), ๋น ๋ฅด๊ฒ ํ๊ณ 2์ ๋๋ฌํ ๊ฒ์
๋๋ค. ๋ฉ๋ชจ๋ฆฌ ๋ฐ ์๋ ด ์ ์ฝ ์กฐ๊ฑด ๋๋ฌธ์ ๋ก์ปฌ ``batch_size=1`` ์กฐ๊ฑด ์กฐ์ฐจ๋ FSDP๋ฅผ ๋จ๋
์ผ๋ก ์ฌ์ฉํ ์ ์์ต๋๋ค. ์๋ฅผ ๋ค์ด, Llama2 ๊ธ๋ก๋ฒ ๋ฐฐ์น ํฌ๊ธฐ๋ 1K์ด๋ฏ๋ก, 2K GPU์์ ์ค์ง ๋ฐ์ดํฐ ๋ณ๋ ฌ ์ฒ๋ฆฌ๋ง์ผ๋ก ์ฌ์ฉ๋ ์ ์์ต๋๋ค. |
|
0 commit comments