4646* ๋ณ๋ ฌํ๋ ๊ณ์ธต( ``nn.Linear `` , ``nn.Embedding `` )์ ์ฐ์ฐ ๋ฐ ๋ฉ๋ชจ๋ฆฌ๋ฅผ ์ ์ฝํ๊ธฐ ์ํด ์ค๋ฉ๋ ์ฐ์ฐ์ ์คํํฉ๋๋ค.
4747
4848ํ
์ ๋ณ๋ ฌ(Tensor Parallel)์ ์ ์ฉํด์ผ ํ๋ ์๊ธฐ์ ์ด์
49- ---------------------------------------------
49+ -----------------------------------------------
5050PyTorch์ Fully Sharded Data Parallel(FSDP)๋ ์ด๋ฏธ ๋ชจ๋ธ ํ์ต์ ํน์ ์์ GPU๋ก ์กฐ์ ํ ์ ์๋ ๊ธฐ๋ฅ์ ๊ฐ์ถ๊ณ ์์ต๋๋ค. ๊ทธ๋ฌ๋, ๋ชจ๋ธ ํฌ๊ธฐ์ GPU ์ ์ธก๋ฉด์์ ๋ชจ๋ธ ํ์ต์ ๋ ํ์ฅํ๋ ค๋ฉด,
5151Tensor Parallel๊ณผ FSDP์ ๊ฒฐํฉ์ด ํ์ํ, ๋ค์๊ณผ ๊ฐ์ ์ถ๊ฐ์ ์ธ ๊ณผ์ ๊ฐ ๋ค์ ๋ฐ์ํ ์ ์์ต๋๋ค.
5252
@@ -61,7 +61,7 @@ Tensor Parallel๊ณผ FSDP์ ๊ฒฐํฉ์ด ํ์ํ, ๋ค์๊ณผ ๊ฐ์ ์ถ๊ฐ์ ์ธ
6161
6262
6363ํ
์ ๋ณ๋ ฌ(Tensor Parallel)์ ์ ์ฉํ๋ ๋ฐฉ๋ฒ
64- ----------------------------
64+ --------------------------------------
6565
6666PyTorch ํ
์ ๋ณ๋ ฌ API๋ ๋ชจ๋ธ์ ๊ฐ ๊ฐ๋ณ ๋ ์ด์ด์ ๋ํ ์ค๋ฉ์ ๊ตฌ์ฑํ๊ธฐ ์ํด ๋ค์๊ณผ ๊ฐ์ ๋ชจ๋ ์์ค์ ์ด์ ์ธํธ (``ParallelStyle ``)๋ฅผ ์ ๊ณตํฉ๋๋ค.
6767
@@ -173,7 +173,7 @@ Llama ๋ชจ๋ธ์ ๊ฒฝ์ฐ, ์ดํ
์
๋ ์ด์ด์์๋ ํํ์ ๊ด๋ จ๋ ์ฌ๋ฌ
173173 ํด๋น ๋ชจ๋ธ์ด ๋๋ฌด ์ปค์ CPU ๋ฉ๋ชจ๋ฆฌ์ ๋ง์ง ์๋ ๊ฒฝ์ฐ, ``meta `` ์ฅ์น ์ด๊ธฐํ (์๋ฅผ ๋ค์ด, ๋ฉํ ์ฅ์น์์ ๋จผ์ ์ด๊ธฐํํ๊ฑฐ๋ ๋ ์ด์ด๋ฅผ ์ค๋ฉํ๊ณ ๋ชจ๋ธ์ ๊ตฌ์ฒดํํ๋ ๋ฑ)๋ฅผ ์ฌ์ฉํ๊ฑฐ๋ ํธ๋์คํฌ๋จธ ๋ชจ๋ธ ์ด๊ธฐํ ์ค์ ``TransformerBlock `` ๋ ์ด์ด๋ฅผ ๋ ์ด์ด๋ณ๋ก ๋ณ๋ ฌํํ ์ ์์ต๋๋ค.
174174
175175``LayerNorm/RMSNorm `` ๋ ์ด์ด์ ์ํ์ค ๋ณ๋ ฌ(Sequence Parallel) ์ ์ฉํ๊ธฐ
176- -------------------------------------------------------
176+ ----------------------------------------------------------------
177177
178178์ํ์ค ๋ณ๋ ฌ(Sequence Parallel)์ ์์ ์ค๋ช
ํ ํ
์ ๋ณ๋ ฌ ์์์ ๋์ํฉ๋๋ค. ๊ธฐ๋ณธ์ ์ธ ํ
์ ๋ณ๋ ฌ์ ``Attention `` ๋ชจ๋๊ณผ ``FeedForward `` ๋ชจ๋ ๋ด์์๋ง ํ
์๋ฅผ ์ค๋ฉํ๊ณ ๋ชจ๋ ์
๋ ฅ๊ณผ ์ถ๋ ฅ (์ฆ, forward pass์ ํ์ฑํ ๋ฐ backward pass์์ ๋ณํ๋)์ ๋ณต์ ๋๋๋ก ์ ์งํ๋ ๊ฒ๊ณผ ๋น๊ตํ ๋, ์ํ์ค ๋ณ๋ ฌ(Sequence Parallel)์ ์ํ์ค ์ฐจ์์์ ์ค๋ฉ๋ ์ํ๋ฅผ ์ ์งํฉ๋๋ค.
179179
@@ -254,7 +254,7 @@ Llama ๋ชจ๋ธ์ ๊ฒฝ์ฐ, ์ดํ
์
๋ ์ด์ด์์๋ ํํ์ ๊ด๋ จ๋ ์ฌ๋ฌ
254254
255255
256256 ์์ค ๋ณ๋ ฌ(Loss Parallel) ์ ์ฉํ๊ธฐ
257- -------------------
257+ -------------------------------
258258
259259์์ค ๋ณ๋ ฌ(Loss Parallel)์ ์์ค ํจ์๋ฅผ ๊ณ์ฐํ ๋ ๋ฉ๋ชจ๋ฆฌ์ ํต์ ์ ์ ์ฝํ๋ ๊ด๋ จ ๊ธฐ์ ๋ก, ์ผ๋ฐ์ ์ผ๋ก ๋ชจ๋ธ ์ถ๋ ฅ์ด ๋งค์ฐ ํฌ๊ธฐ ๋๋ฌธ์ ์ฌ์ฉํฉ๋๋ค. ์์ค ๋ณ๋ ฌ์์๋ ๋ชจ๋ธ ์ถ๋ ฅ์ด (์์ฃผ ๊ฑฐ๋ํ) ์ดํ ์ฐจ์์์ ์ค๋ฉ๋ ๋, ๋ชจ๋ ๋ชจ๋ธ ์ถ๋ ฅ์ ๋งค๋ฒ ๋จ์ผ GPU์ ๋ชจ์ผ์ง ์๊ณ ๋ ๊ต์ฐจ ์ํธ๋กํผ ์์ค์ ํจ์จ์ ์ผ๋ก ๊ณ์ฐํ ์ ์์ต๋๋ค. ์ด๋ ๋ฉ๋ชจ๋ฆฌ ์๋น๋ฅผ ์ ์ํ๊ฒ ์ค์ผ ๋ฟ๋ง ์๋๋ผ, ํต์ ์ค๋ฒํค๋๋ฅผ ์ค์ด๊ณ ์ค๋ฉ๋ ์ฐ์ฐ์ ๋ณ๋ ฌ๋ก ์ฒ๋ฆฌํ์ฌ ํ์ต ์๋๋ฅผ ๊ฐ์ ํฉ๋๋ค. ์๋ ๊ทธ๋ฆผ์ ์์ค ๋ณ๋ ฌ์ด ์ค๋ฉ๋ ์ฐ์ฐ์ ํตํด ๋จ์ผ GPU๋ง๋ค ๋ชจ๋ ๋ชจ๋ธ์ ์ถ๋ ฅ์ ๋ชจ์ผ๋ ๊ฒ์ ํผํ๋ ๋ฐฉ๋ฒ์ ๊ฐ๋ตํ ๋ณด์ฌ์ค๋๋ค.
260260
0 commit comments