Xiga xigaHagaha xiga
Doorashada GPU ee Barashada Mashiinka
Farsamo
HAGAHA Farsamada
Triton waa luuqad ku salaysan Python iyo soo-sameeyaha loogu talagalay qorista kernels GPU-da marka loo eego blocks ee curiyeyaasha halkii ay ka ahaan lahayd dunta hardware-ga gaarka ah.
Dhageyso hagahanAI-waa laga sheekeeyay · ~3 min
Waxay fududayn kartaa hawlaha tensor-ka caadiga ah marka la barbar dhigo CUDA cayriin, iyadoo weli u baahan in si taxadar leh loo xakameeyo gelitaanka xusuusta, waji-xidhka, bilaabista joomatari iyo waxqabadka gaarka ah ee qalabka.
Barnaamij-samaynta GPU-gu wuxuu badiyaa u baahan yahay shaqada qaybinta dunta badan iyadoo la maareynayo dhaqdhaqaaqa xusuusta. Triton waxay bixisaa Python ku salaysan luuqad gaar ah oo domain ah kaas oo kernel uu ku qeexayo hawlgallada ka sarreeya qiyamka. Hawsha loo qurxiyey ururinta Triton waxay isticmaali kartaa dhismayaal sida aqoonsiga barnaamijka si loo aqoonsado xannibaad iyo kala duwanaansho si loo abuuro curiyeyaasha. Soo-sameeyaha ayaa khariidado ka dhigaya barnaamijkan heerka xannibaadda ee fulinta GPU, taasoo u oggolaanaysa horumariyayaasha inay qoraan hawlgallada gaarka ah iyaga oo aan cayimin tilmaamaha dunta kasta sida CUDA C++ caadiga ah. Isku-darka aasaasiga ah ee vector-ka ayaa u qaybiya vector dheer oo u kala qaybiya blocks. Barnaamij kasta oo Triton ah waxa uu qabtaa hal baloog aqoonsiga, xisaabiya dhimista, rarka qiyamka labada qalab, ku dara oo kaydiyaa wax soo saarka. Qeybta ugu dambeysa ayaa laga yaabaa inay dhaafto dhererka tensor-ka saxda ah, si ay wejiyada u ilaaliso rarka iyo kaydinta. Maaskarada aan saxda ahayn waxay akhrin kartaa xusuusta aan shaqaynayn ama waxay ka tagi kartaa qiyam sax ah. Bilaw habaynta, oo ay ku jiraan tirada warps iyo cabbirka xannibaadda, waxay saamaysaa ururinta iyo fulinta. Hawlgallada matrixka, tillaabtu waxay ku hayn kartaa xogta si degdeg ah xusuusta-chip-ka waxayna dib u isticmaali kartaa qiyamka xisaabinta. Tani waxay wanaajisaa qaababka dhaqdhaqaaqa xogta, laakiin cabbirada foornada waa inay ku habboon yihiin agabka qalabka waxayna u dhaqmi karaan si ka duwan qaababka wax gelinta. Isku-duwaha Triton wuxuu taageeraa hagaajinta iyo hab-socodka shaqada iswada, haddana kernel-ku si toos ah ugama dhaqsiyo hawlgalka maktabadda la hagaajiyay. Waxyaalaha yar yar waxa laga yaabaa inay u badan tahay bilowga sareynta, wakhtiga isku-dubbaridkana waa inaan lagu khaldin wakhtiga shaqo ee dawlad-goboleedka ah. Samaynta kernel waxay u baahan tahay hubinta saxda ah ee qaababka, talaabooyinka, noocyada iyo qalabka. Is barbar dhig ka soo horjeedka fulinta la aamini karo iyadoo la isticmaalayo dulqaad tirooyin ku habboon xisaabinta dul sabaynaysa. Benchmark leh diirran, isku-dubarid iyo culaysyo shaqo oo wakiil ah. Hubi koodka la soo saaray ama raadiyaha profiler marka waxqabadku ka duwan yahay filashooyinka. Triton waxay yaraynaysaa cadadka kuleyliyaha heerka hoose; kama saarayso baahida loo qabo in la fahmo isugaynta xusuusta, deggenaanshaha, ganacsiga saxda ah ama xaaladaha jinsiyadeed. Adeegso marka hawlgalka isku dhafan ee caadadii ah ama qaab khaas ah uu caddeeyo kharashka dayactirka, oo uu haysto dib u dhac la isku halayn karo marka qalabka iyo kombuyuutarku ay kala duwan yihiin.
Malo-awaal ah oo ku darida kernel-ka ayaa tusaale ahaan barnaamij kasta u dhigaya qayb-indices ah, waxay ku rartay laba baloog oo maaskaro dabada ah, ku dara oo kaydisa natiijada.
Tababbarrada isku dhufashada matrix-ku waxa ay gelisaa jaangooyooyin-blocks si xogta dib loo isticmaali karo, iyada oo la dhimayo taraafikada xusuusta ee soo noqnoqda marka la barbar dhigo habka curiye-cunsurka.
Horumariyahu waxa uu isbarbar dhigayaa kernel Triton iyo hawlgalka PyTorch ee qaababka wakiilada waxana ku jira isku dubarid iyo isku-dubarid habka wakhtiga.
Kernel-ku wuxuu gacanta ku hayaa cabbirka tensor-ka oo aan loo qaybin cabbirka xannibaadda isagoo qarinaya rarka iyo bakhaarada bannaanka, isagoo ka hortagaya gelitaanka xusuusta aan ansax ahayn ee barnaamijka ugu dambeeya.
Triton waa mid faa'iido leh marka kooxuhu u baahan yihiin hawlgal GPU gaar ah oo ay ilaalin karaan koodka qalabka gaarka ah. Socod-hawleed la taaban karo ayaa ku bilaabmaya sal-dhis heer sare ah oo sax ah, waxa uu ku daraa kernel kaliya ka dib marka sifayntu aqoonsato dhuunta dhalo, oo ay tijaabiso qaababka wakiilada iyo kiisaska geesaha. Warbixinada hawlqabadku waa inay ku jiraan diirimaad, qalab, dtype iyo qaabaynta bilowga si natiijooyinka dib loo soo saari karo. Hagaajinta ururiyaha ayaa laga yaabaa inay balaadhiso fursadaha hagaajinta, laakiin kooxuhu wali waxay u baahan yihiin dariiqyada dib u dhaca iyo hubinta nooca. Go'aanka ugu muhiimsan waa haddii xawaaraha kernel-ka caadiga ah ama faa'iidada isku dhafka ah ay caddeyso tijaabinta iyo kharashka dayactirka.
Warqad bilaash ah
Saddex sheeko oo la xaqiijiyay oo AI ah maalin kasta oo shaqo ah, oo lagu qoray Ingiriis fudud. Bilaash had iyo jeer, aan xayaysiis lahayn.
Hal iimayl maalin kasta oo shaqo ah. Ka bax hal gujin. Marna ma iibinno ama ma wadaagno ciwaankaaga.
Is tijaabi
Fikrad degdeg ah oo ku saabsan jawaab kasta, iyo shahaado la wadaajin karo oo leh aqoonsi la xaqiijin karo markaad ka gudubto koorso.
Taageer waxbarashada AI bilaashka ah. AI Understanding waa 501(c)(3) aan faa'iido doon ahayn — xayaysiis la'aan, albaab lacag ah ma jiro, had iyo jeer. Samee deeq
Triton waa luuqad ku salaysan Python iyo soo-sameeyaha loogu talagalay qorista kernels GPU-da marka loo eego blocks ee curiyeyaasha halkii ay ka ahaan lahayd dunta hardware-ga gaarka ah. Waxay fududayn kartaa hawlaha tensor-ka caadiga ah marka la barbar dhigo CUDA cayriin, iyadoo weli u baahan in si taxadar leh loo xakameeyo gelitaanka xusuusta, waji-xidhka, bilaabista joomatari iyo waxqabadka gaarka ah ee qalabka.
Triton waxay daaha ka qaadaysaa soo saarista borogaraamyada heerka xannibaadda ah halka khariidadaha soo-ururinta ay ka shaqeeyaan fulinta GPU.
Qeybta ugu dambeysa waxay dhaafin kartaa cabbirka qaabka macquulka ah, markaa waji-xidhaha ayaa ka hortagaya gelitaanka iyo dukaamada aan sax ahayn.
Aqoonsiga barnaamijka ayaa u oggolaanaya kernel-ku inuu xisaabiyo qaybta wax soo saarka ee ay tahay in uu habeeyo.
Jeexdintu waxay dib u isticmaali kartaa xogta lagu hayo xusuusta degdegga ah halkii ay si isdaba joog ah ugu shubi lahayd hawlgal kasta oo xisaabeed.
Isku-dubaridku waxa ay maamuli karaan wicitaanka ugu horreeya, markaa waxqabadka gobolka ee joogtada ah waa in si gaar ah loo cabbiraa marka taasi tahay isbarbardhigga loogu talagalay.
Sii wad waxbarashada
Tilmaamayaal badan ayaa loo doortay mawduucan
Xiga xigaHagaha xiga
Doorashada GPU ee Barashada Mashiinka
Farsamo