Ntụziaka nka

Usoro Ndekota na mgbanaka Ntị

Usoro myirịta na-ekewa otu usoro ntinye ogologo n'ofe ọtụtụ GPUs n'akụkụ akara (oge), yana Ntị na-eme ka GPU ndị ahụ gbakọọ nlebara anya kpọmkwem site n'ịfefe igodo/ọnụ ahịa gburugburu mgbanaka.

Nchịkọta

Usoro myirịta na-ekewa otu usoro ntinye ogologo n'ofe ọtụtụ GPUs n'akụkụ akara (oge), yana Ntị na-eme ka GPU ndị ahụ gbakọọ nlebara anya kpọmkwem site n'ịfefe igodo/ọnụ ahịa gburugburu mgbanaka. Ha jikọtara ọnụ na-eme ka nde-token okirikiri windo ga-ekwe omume na-enweghị otu GPU na-ejide usoro niile.

Usoro Parallelism na nlebara anya mgbanaka bụ ngọngọ ụlọ ọrụ nka nke na-emetụta ịdịmma nlereanya, ọnụ ahịa akụrụngwa, nkwụsịtụ, na ntụkwasị obi n'ogo.

Ime miri emi

Nlebara anya ọkọlọtọ chọrọ ajụjụ ọ bụla iji hụ igodo / uru ọ bụla, yabụ ebe nchekwa ịgbalite na-eto n'ogologo usoro yana K/V zuru ezu ga-adịrịrị. Usoro myirịta na-emebi usoro nke mere GPU nke ọ bụla nwere ntụtụ akara ngosi (na ajụjụ ha, igodo, ụkpụrụ). Ntị mgbanaka na-ahazi GPU na mgbanaka ezi uche dị na ya: ngwaọrụ ọ bụla na-edobe ajụjụ mpaghara ya ebe a na-agafe ihe mgbochi K/V hop-by-hop gburugburu mgbanaka ahụ. Dị ka ngọngọ ọ bụla na-abịarute, GPU na-agbakọ nlebara anya nke ọma ma na-achịkọta nsonaazụ site na iji online-softmax (otu aghụghọ max/nchikota dị ka FlashAttention). Ka emechara loop zuru oke, ajụjụ ọ bụla abanyela na igodo ọ bụla, na-enweghị GPU na-echekwa K/V dum. N'ụzọ dị mkpa, nkwurịta okwu K/V na-ejikọta na mgbakọ, ya mere ọ na-agbakwụnye obere ego elekere mgbidi.

Nghọta nka nka

Ntị mgbanaka na-adabere na softmax dị n'ịntanetị: enwere ike ịgbakọ nlebara anya ngọngọ-site-block ka ị na-edobe oke na-agba ọsọ yana ihe na-eme ihe na-eme ka ọ na-agba ọsọ, wee na-atụgharị nchikota akụkụ akụkụ mbụ mgbe uru ka ukwuu pụtara. Nke a na-eme ka nsonaazụ mgbakọ na mwepụ yitere na nlebara anya zuru oke. Mgbanaka ahụ na-agafe naanị K / V tenors (nha nha na ngọngọ, ọ bụghị usoro zuru ezu), na n'ihi na nkwurịta okwu hop ọ bụla na-ekpuchi matmul ngọngọ gara aga, bandwit - ọ bụghị ebe nchekwa - na-aghọ ihe na-egbochi ya.

Nleba anya na mgbanaka usoro nhazi

Usoro myirịta na-ekewa otu usoro ntinye ogologo n'ofe ọtụtụ GPUs n'akụkụ akara (oge), yana Ntị na-eme ka GPU ndị ahụ gbakọọ nlebara anya kpọmkwem site n'ịfefe igodo/ọnụ ahịa gburugburu mgbanaka. Ha jikọtara ọnụ na-eme ka nde-token okirikiri windo ga-ekwe omume na-enweghị otu GPU na-ejide usoro niile. Usoro Parallelism na nlebara anya mgbanaka bụ ngọngọ ụlọ ọrụ nka nke na-emetụta ịdịmma nlereanya, ọnụ ahịa akụrụngwa, nkwụsịtụ, na ntụkwasị obi n'ogo. Iji wulite nghọta miri emi, na-emeso usoro Parallelism na mgbanaka nleba anya dị ka ihe nlere anya na-arụ ọrụ, ọ bụghị otu njirimara: kọwaa nsonaazụ achọrọ, dokwuo anya echiche, wee kewaa ihe sistemụ nwere ike ime nke ọma na ihe ka chọrọ mkpebi ndị ọkachamara.

Na omume, otu ndị siri ike na-eji usoro Parallelism na mgbanaka nlebara anya na-ebuli ụkpụrụ ụlọ, data na akụrụngwa megide ntụkwasị obi na ọnụ ahịa. Ha na-edepụta njirisi ịga nke ọma nke ọma, nwalee megide data ziri ezi yana usoro ọrụ, yana na-atụgharị dabere na usoro ọdịda ahụrụ karịa karịa mmeri otu oge. Nke a bụ ebe nghọta usoro ihe atụ na-atụgharị ghọọ ike na-adịgide adịgide n'ofe ngwaahịa, amụma na arụmọrụ.

Mkpebi ihe owuwu ụlọ na-akwalite arụmọrụ yana ọnụ ahịa ọrụ ruo ọtụtụ afọ. N'otu oge ahụ, ịkwalite otu akara ngosi nwere ike zoo adịghị ike sistemụ sara mbara. Ụzọ kachasị na-agbanwe agbanwe bụ ijikọ ọsọ nnwale na ịdọ aka ná ntị ọchịchị: ndị na-anya ụgbọ elu, ijide ihe akaebe, bipụta ndekọ mkpebi, na na-aga n'ihu na-emelite nchekwa dị ka omume nlereanya, atụmanya ndị ọrụ, na ihe iwu chọrọ.

Mmetụta atụmatụ

Mkpebi ihe owuwu ụlọ na-akwalite arụmọrụ yana ọnụ ahịa ọrụ ruo ọtụtụ afọ.

Mkpebi ihe owuwu ụlọ na-akwalite arụmọrụ yana ọnụ ahịa ọrụ ruo ọtụtụ afọ. N'ịkwanye ọkwa dị elu, a na-atụgharị nke a ka ọ bụrụ iwu arụ ọrụ enwere ike ịtụnye, oke nwe, na emume ntụlegharị ugboro ugboro ka ndị otu wee nwee ike ịbawanye ntụkwasị obi kama iwelite enweghị mgbagha.

Nkà mmụta nka na-enyere ndị otu egwuregwu aka ịhọrọ nchịkọta ziri ezi, ọ bụghị naanị nke kachasị ọhụrụ.

Nkà mmụta nka na-enyere ndị otu egwuregwu aka ịhọrọ nchịkọta ziri ezi, ọ bụghị naanị nke kachasị ọhụrụ. N'ịkwanye ọkwa dị elu, a na-atụgharị nke a ka ọ bụrụ iwu arụ ọrụ enwere ike ịtụnye, oke nwe, na emume ntụlegharị ugboro ugboro ka ndị otu wee nwee ike ịbawanye ntụkwasị obi kama iwelite enweghị mgbagha.

Nhọrọ injinia ka mma na-ebelata ihe omume ntụkwasị obi na mmepụta.

Nhọrọ injinia ka mma na-ebelata ihe omume ntụkwasị obi na mmepụta. N'ịkwanye ọkwa dị elu, a na-atụgharị nke a ka ọ bụrụ iwu arụ ọrụ enwere ike ịtụnye, oke nwe, na emume ntụlegharị ugboro ugboro ka ndị otu wee nwee ike ịbawanye ntụkwasị obi kama iwelite enweghị mgbagha.

Ọdịnihu nke usoro ndakọrịta na mgbanaka nlebara anya

Myirịta usoro na-aghọ ọkọlọtọ maka ọzụzụ na ntinye okwu ogologo oge, na-ejikọta ya na myirịta tensor na pipeline n'ime okirikiri nhọrọ ukwuu '4D' ma ọ bụ '5D'. Ndị dị iche iche dị ka straipu ma ọ bụ zigzag nlebara anya na-eweghachi ọrụ nke ihe mkpuchi ihe kpatara. Na-atụ anya mgbanaka topology maara nke ọma na NVLink yana njikọta siri ike na mwepu KV-cache, na-akwalite ogologo ọnọdụ gburugburu ruo ọtụtụ iri nde token maka iweghachite, codebases na ogologo akwụkwọ.

Mmejuputa n'ezie n'ụwa

Ọzụzụ okirikiri 1M-token LLM site na iji Ntị mgbanaka kesaa usoro nke ọ bụla n'ofe 8 GPU.

Usoro nke Megatron-LM na-ebelata ebe nchekwa ọrụ na LayerNorm na mpaghara nkwụsị

Na-ahazi akwụkwọ dum ma ọ bụ nnukwu ebe nchekwa koodu n'otu ngafe na-aga n'ihu na-enweghị mkpọpu

Ijikọta nleba anya mgbanaka na myirịta tensor ka dabara na ntụnye okwu ogologo ogologo na ọnụ ọtụtụ GPU

Usoro mmejuputa

Usoro ndakọrịta na mgbanaka nlebara anya na omume

Ọzụzụ okirikiri 1M-token LLM site na iji Ntị mgbanaka na-ekekọrịta usoro ọ bụla n'ofe 8 GPU.

Ọzụzụ ihe ngosi 1M-token LLM site n'ịkọwa usoro nke ọ bụla n'ofe 8 GPU na ndị otu Nleba anya mgbanaka na-enwetakarị nsonaazụ kacha mma mgbe ha kọwapụtara ọnụ ụzọ dị mma n'ihu, debe ụzọ mmụba mmadụ maka ikpe ọnụ, wee soro ma uru nrụpụta yana ụgwọ njehie n'ime oge.

Usoro ndakọrịta na mgbanaka nlebara anya na omume

Usoro nke Megatron-LM na-ebelata ebe nchekwa ọrụ na LayerNorm na mpaghara nkwụsị.

Usoro nke Megatron-LM na-ebelata ebe nchekwa ịgbalite na LayerNorm na mpaghara ndị na-apụ apụ na-enwetakarị nsonaazụ ka mma mgbe ha na-akọwapụta ọnụ ụzọ dị mma n'ihu, na-eme ka ụzọ mmadụ si abawanye maka ọnụ okwu, ma soro ma uru mmepụta na ụgwọ njehie na oge.

Usoro ndakọrịta na mgbanaka nlebara anya na omume

Na-ahazi akwụkwọ dum ma ọ bụ nnukwu ebe nchekwa koodu n'otu ngafe na-aga n'ihu na-enweghị mkpọpu.

Ịhazi akwụkwọ dum ma ọ bụ nnukwu koodu nchekwa n'otu ngafe na-aga n'ihu na-enweghị mkpọpu otu ìgwè na-enwetakarị nsonaazụ ka mma mgbe ha na-akọwapụta ọnụ ụzọ dị mma n'ihu, na-eme ka ụzọ mmadụ na-ebuwanye ibu maka ikpe ikpe, ma soro ma uru nrụpụta na ụgwọ njehie na oge.

Usoro ndakọrịta na mgbanaka nlebara anya na omume

Ijikọta nleba anya mgbanaka na myirịta tensor ka ọ dabara na ntinye okwu ogologo ogologo na ọnụ ọtụtụ GPU.

Ijikọ nleba anya mgbanaka na tensor parallelism ka dabara adaba ogologo oge okwu na ọnụ ọgụgụ GPU dị iche iche Otu egwuregwu na-enwetakarị nsonaazụ kacha mma mgbe ha kọwapụtara ọnụ ụzọ dị mma n'ihu, debe ụzọ mmụba mmadụ maka ikpe ọnụ, wee soro ma uru nrụpụta yana ụgwọ njehie n'ime oge.

Ihe ize ndụ & okporo ụzọ nche

!

Ịkwalite otu akara ngosi nwere ike zoo adịghị ike sistemụ sara mbara.

!

A na-eledakarị ihe akụrụngwa na ụgwọ ọrụ anya.

!

Ọdịiche nchekwa na nleba anya nwere ike itolite ka sistemu na-adịwanye mgbagwoju anya.

Map mmejuputa

1

Kọwaa latency, ịdịmma na ebumnuche ọnụ ahịa tupu mmejuputa ya.

Kọwaa latency, ịdịmma na ebumnuche ọnụ ahịa tupu mmejuputa ya. Mesoo nzọụkwụ ọ bụla dị ka ọnụ ụzọ akaebe: ọ bụrụ na emezughị ụkpụrụ, kwụsịtụ mbugharị, mechie oghere ahụ, naanị wee gbasaa ojiji.

2

Benchmark n'okpuru ibu dị adị na ọnọdụ data.

Benchmark n'okpuru ibu dị adị na ọnọdụ data. Mesoo nzọụkwụ ọ bụla dị ka ọnụ ụzọ akaebe: ọ bụrụ na emezughị ụkpụrụ, kwụsịtụ mbugharị, mechie oghere ahụ, naanị wee gbasaa ojiji.

3

Nleba anya akụrụngwa maka mperi, ịkpafu na mmetụta onye ọrụ.

Nleba anya akụrụngwa maka mperi, ịkpafu na mmetụta onye ọrụ. Mesoo nzọụkwụ ọ bụla dị ka ọnụ ụzọ akaebe: ọ bụrụ na emezughị ụkpụrụ, kwụsịtụ mbugharị, mechie oghere ahụ, naanị wee gbasaa ojiji.

4

Kwadebe ụzọ nzaghachi azụghachi azụ na ihe omume tupu ịchachaa.

Kwadebe ụzọ nzaghachi azụghachi azụ na ihe omume tupu ịchachaa. Mesoo nzọụkwụ ọ bụla dị ka ọnụ ụzọ akaebe: ọ bụrụ na emezughị ụkpụrụ, kwụsịtụ mbugharị, mechie oghere ahụ, naanị wee gbasaa ojiji.

Nọgide na-eme nchọpụta