Ubuyobozi bwa tekiniki

Icyitonderwa

Flash Attention nuburyo bwubwenge bwo kubara intambwe yimbere muri Transformers utarigeze wandika matrix nini yo kwitondera buhoro buhoro.

2 min somaIbiherutse kuvugururwa

Incamake

It makes long-context models far faster and more memory-efficient without changing their math.

Kwibira cyane

Icyitonderwa gisanzwe kigereranya buri kimenyetso nikindi kimenyetso, gitanga amanota ya N-by-N ikura inshuro enye hamwe nuburebure bwikurikiranya. Na Naive, iyo matrix yandikiwe kandi isomwe uhereye muri GPU yibuka ryinshi rya HBM (HBM), kandi ko guhinduranya - ntabwo ari kugwira - nicyo cyuho. Flash Attention, yatangijwe na Tri Dao na bagenzi be muri 2022, itunganya ibarura kugirango matrix itabikwa neza. Itunganya ibibazo, urufunguzo, nagaciro mumashusho mato ahuye byihuse kuri chip SRAM, ibara ibisubizo byigice, kandi ikabidoda hamwe ikoresheje interineti ikora-softmax trick. Ibisohoka ni imibare isa nibisanzwe ariko ikoresha umurongo wibuke kandi ikora inshuro nyinshi byihuse, cyane cyane kumurongo muremure.

Ubushishozi

Amayeri yingenzi ni tiling wongeyeho softmax kumurongo. Ubusanzwe Softmax ikenera umurongo wose wamanota kugirango ibare icyerekezo cyayo, ariko Flash Attention ikomeza gukora ntarengwa kandi ikora amafaranga uko itemba kuri tile, ikuraho ibyasohotse mbere igice kuburyo ibisubizo byanyuma aribyo. Kuberako amanota aringaniye aguma muri SRAM (ordre yubunini bwihuse kurenza HBM), algorithm irabizi IO: igabanya kwibuka gusoma no kwandika aho gukora imibare mibare.

Ingaruka z'Ingamba

Igiciro na bije

Ibyemezo byubwubatsi bitwara imikorere nigiciro cyimikorere kumyaka.

Ibyemezo bisobanutse

Ubuhanga bwa tekinike bufasha amakipe guhitamo umurongo ukwiye, ntabwo ari shyashya gusa.

Kugenzura ubuziranenge

Guhitamo neza bya injeniyeri bigabanya ibintu byizewe mubikorwa.

Ejo hazaza ha Flash

Flash Attention yahindutse inyubako idasanzwe, hamwe na FlashAttention-2 na FlashAttention-3 ikanda byinshi byinjira muri GPU nshya nka H100 mugutezimbere igabana ryakazi no gukoresha inzira nke za FP8. Witege ko uzakomeza gushushanya hamwe nibikoresho, guhuza imbaraga mumahugurwa no gufata umwanzuro, hamwe nibihinduka byahujwe gake, kunyerera-idirishya, hamwe-birebire-by-ibitekerezo. Nkuko imiterere ya Windows irambuye kuri miriyoni yikimenyetso, IO-izi intangiriro nkiyi ikomeza kuba ingenzi kugirango ukomeze kwibuka no kwihuta mubikorwa.

Gushyira mu bikorwa Isi

Guhugura imiterere yindimi nini nka Llama na GPT-urwego rwa sisitemu hamwe na Windows ndende ndende ku giciro cyo kwibuka gito.

Gukorera abafasha kuganira byihuse mukwihutisha icyiciro cyambere aho isomwa rirerire ryambere.

Gushoboza inyandiko-isesengura ibikoresho byinjiza ibitabo byose cyangwa codebase mugukora ibitekerezo birebire byashoboka kuri GPU imwe.

Imbaraga zicyerekezo hamwe namajwi Guhindura amajwi aho ibisubizo bihanitse byinjiza birema ibimenyetso birebire cyane.

Ingaruka & Kurinda

Gutezimbere igipimo kimwe gishobora guhisha intege nke za sisitemu.

Ibikorwa Remezo no kubungabunga akenshi usanga bidahabwa agaciro.

Icyuho cyumutekano no kwitegereza birashobora kwiyongera uko sisitemu igenda igorana.

Igishushanyo mbonera

1

Sobanura ubukererwe, ubuziranenge, nigiciro cyibiciro mbere yo kubishyira mubikorwa.

2

Ibipimo byerekana umutwaro ufatika hamwe namakuru yimiterere.

3

Gukurikirana ibikoresho kubikosa, drift, ningaruka zabakoresha.

4

Tegura inzira yo gusubiza ibyabaye mbere yo gupima.

Komeza Ubushakashatsi

Free newsletter

Get the daily AI briefing

Three verified AI stories every weekday morning, written in plain English. Free forever, no ads.

One email each weekday. Unsubscribe in one click. We never sell or share your address.

Test yourself

Take the Flash Attention quiz

Instant feedback on every answer, and a shareable certificate with a verifiable ID once you pass a course.

Tangira ikibazo

Support free AI education. AI Understanding is a 501(c)(3) nonprofit — no ads, no paywall, ever. Make a donation

Ubuyobozi bukurikira

Itondere Kuzunguruka no Gukata Umutwe

Ibibazo bikunze kubazwa

What is Flash Attention?

Flash Attention nuburyo bwubwenge bwo kubara intambwe yimbere muri Transformers utarigeze wandika matrix nini yo kwitondera buhoro buhoro. Cyakora imiterere-ndende-moderi yihuta cyane kandi yibuka-neza idahinduye imibare.

Ni ubuhe butumwa bukomeye intego ya Flash Attention igamije?

Flash Attention ni IO-ibizi: igabanya amakuru yahinduwe hagati yihuta kuri chip SRAM hamwe nububiko bwihuta bwihuta bwibikoresho, aribwo buryo bworoshye aho kuba imibare ubwayo.

Nigute Flash Attention yirinda kubika matrike yuzuye ya N-by-N?

Itondekanya kubara kuburyo buri gice gihuye na SRAM yihuse, kubara no kwegeranya ibisubizo byigice bitarigeze bihinduka matrix yose muri HBM.

Nubuhe buryo butuma Flash Attention ibara softmax neza utabonye umurongo wose icyarimwe?

Kumurongo wa softmax ikomeza gukora ntarengwa kandi ikora mugihe utondekanya amabati, ukuraho ibisubizo byigice cyambere kugirango ibisanzwe byanyuma birasobanutse neza.

Kuki Flash Attention ifasha cyane hamwe nurutonde rurerure?

Umunzani wo kwitondera matrix nka N-kwaduka murwibutso, bityo rero kwirinda ububiko bwuzuye butanga amafaranga menshi yo kuzigama neza mugihe urutonde ari rurerure.

Niki 'IO-izi' igishushanyo cya Flash Attention ishyira imbere kugabanya?

IO-izi bivuze ko algorithm yateguwe hafi yikiguzi cyo kwimura amakuru murwego rwo kwibuka, kugabanya traffic HBM kuruta ibikorwa byimibare.