Èdè AI Itọsọna

FlashAttention

FlashAttention jẹ alugoridimu-daradara-iranti ti o ṣe iṣiro akiyesi deede kanna bi awọn Ayirapada boṣewa ṣugbọn laisi kikọ matrix akiyesi omiran nigbagbogbo lati fa fifalẹ iranti GPU.

Akopọ

FlashAttention jẹ alugoridimu-daradara-iranti ti o ṣe iṣiro akiyesi deede kanna bi awọn Ayirapada boṣewa ṣugbọn laisi kikọ matrix akiyesi omiran nigbagbogbo lati fa fifalẹ iranti GPU. O ṣe ikẹkọ ọrọ-ọrọ gigun ati itọkasi ni iyara yiyara ati din owo.

FlashAttention jẹ apakan ti akopọ ede-AI ti a lo lati ka, ṣe ipilẹṣẹ, ṣe lẹtọ, ati yi ọrọ ati ọrọ pada ni iwọn.

Jin Dive

Ifarabalẹ deede ṣe iṣiro Dimegilio kan fun gbogbo bata ti awọn ami, ti n ṣe agbejade matrix N-by-N. Fun ọkọọkan 4,000-àmi ti o jẹ awọn ikun miliọnu 16, ati pe matrix gbọdọ wa ni kikọ si ati ka pada lati iranti bandiwidi giga ti GPU (HBM). Ijabọ iranti yẹn, kii ṣe iṣiro, jẹ igo gidi. FlashAttention, ti a ṣe nipasẹ Tri Dao ati awọn ẹlẹgbẹ ni 2022, tun ṣe iṣiro naa ki matrix naa ko ni ohun elo ni kikun. O ṣe ilana ọkọọkan ni awọn alẹmọ ti o baamu ni kekere GPU, SRAM iyara-yara lori chip, ṣiṣe iṣiro softmax ni afikun bi o ti n lọ. Abajade jẹ aami mathematiki si akiyesi boṣewa ṣugbọn o nlo iranti ti o kere pupọ ati ṣiṣe ni iyara pupọ, ti n mu awọn window ọrọ to gun to gun.

Imọ-imọ-ẹrọ

Awọn omoluabi ni 'online softmax' ni idapo pelu tiling. FlashAttention ṣe ẹru awọn bulọọki kekere ti awọn ibeere, awọn bọtini, ati awọn iye sinu SRAM, ṣe iṣiro awọn abajade ifarabalẹ apakan, ati ṣe atunṣe awọn akopọ ṣiṣe bi awọn bulọọki tuntun ti de nitoribẹẹ imudara softmax deede duro laisi ri gbogbo awọn ikun ni ẹẹkan. Nitoripe ko tọju matrix N-by-N ni kikun ni HBM, awọn iwọn iranti laini kuku ju quadratically, ati ekuro ti dapọ sinu iṣẹ GPU kan lati dinku iranti kika ati kikọ.

Titunto si FlashAttention

FlashAttention jẹ alugoridimu-daradara-iranti ti o ṣe iṣiro akiyesi deede kanna bi awọn Ayirapada boṣewa ṣugbọn laisi kikọ matrix akiyesi omiran nigbagbogbo lati fa fifalẹ iranti GPU. O ṣe ikẹkọ ọrọ-ọrọ gigun ati itọkasi ni iyara yiyara ati din owo. FlashAttention jẹ apakan ti akopọ ede-AI ti a lo lati ka, ṣe ipilẹṣẹ, ṣe lẹtọ, ati yi ọrọ ati ọrọ pada ni iwọn. Lati kọ oye ti o jinlẹ, tọju FlashAttention bi awoṣe iṣẹ, kii ṣe ẹya ẹyọkan: ṣalaye awọn abajade ti o fẹ, ṣe alaye awọn arosinu, ati sọtọ ohun ti eto le ṣe ni igbẹkẹle lati ohun ti o tun nilo idajọ amoye.

Ni iṣe, awọn ẹgbẹ ti o lagbara ni lilo awọn itọsi apẹrẹ FlashAttention, imupadabọ, ati atunyẹwo awọn losiwajulosehin bi eto ibaraẹnisọrọ iṣọpọ kan. Wọn ṣe akọsilẹ awọn ibeere aṣeyọri ti o fojuhan, idanwo lodi si data ojulowo ati ṣiṣan iṣẹ, ati atunbere ti o da lori awọn ilana ikuna ti a ṣakiyesi dipo awọn bori ala-akoko kan. Eyi ni ibiti oye imọ-jinlẹ yipada si agbara ti o tọ kọja ọja, eto imulo, ati awọn iṣẹ ṣiṣe.

Ṣiṣan iṣẹ ede le gbe ni iyara laisi irubọ aitasera. Ni akoko kanna, awọn otitọ hallucinated le tẹ awọn ijabọ laiparuwo, awọn ṣiṣan atilẹyin, tabi awọn abajade iwadii. Ọna resilient julọ julọ ni lati darapọ iyara idanwo pẹlu ibawi ijọba: ṣiṣe awọn awakọ awakọ, mu ẹri mu, ṣe atẹjade awọn iwe ipinnu, ati imudojuiwọn awọn aabo nigbagbogbo bi ihuwasi awoṣe, awọn ireti olumulo, ati awọn ibeere ilana ti dagbasoke.

Ipa Ilana

Ṣiṣan iṣẹ ede le gbe ni iyara laisi irubọ aitasera.

Ṣiṣan iṣẹ ede le gbe ni iyara laisi irubọ aitasera. Ni awọn imuṣiṣẹ ti o ni agbara giga, eyi ni a tumọ si awọn ofin iṣiṣẹ wiwọn, awọn aala nini, ati awọn ilana atunyẹwo loorekoore ki awọn ẹgbẹ le ṣe iwọn igbẹkẹle dipo iwọn aibikita.

O faagun iraye si kọja awọn ede ati awọn aza ibaraẹnisọrọ.

O faagun iraye si kọja awọn ede ati awọn aza ibaraẹnisọrọ. Ni awọn imuṣiṣẹ ti o ni agbara giga, eyi ni a tumọ si awọn ofin iṣiṣẹ wiwọn, awọn aala nini, ati awọn ilana atunyẹwo loorekoore ki awọn ẹgbẹ le ṣe iwọn igbẹkẹle dipo iwọn aibikita.

Awọn ẹgbẹ le lo akoko diẹ sii lori idajọ lakoko ti adaṣe n kapa atunwi.

Awọn ẹgbẹ le lo akoko diẹ sii lori idajọ lakoko ti adaṣe n kapa atunwi. Ni awọn imuṣiṣẹ ti o ni agbara giga, eyi ni a tumọ si awọn ofin iṣiṣẹ wiwọn, awọn aala nini, ati awọn ilana atunyẹwo loorekoore ki awọn ẹgbẹ le ṣe iwọn igbẹkẹle dipo iwọn aibikita.

Ojo iwaju ti FlashAttention

FlashAttention ti di bulọọki ile aiyipada. FlashAttention-2 ṣe ilọsiwaju pipin iṣẹ GPU, ati FlashAttention-3 lo awọn ẹya ohun elo ohun elo Hopper tuntun bii asynchrony ati FP8 konge kekere. Reti ti o tẹsiwaju-apẹrẹ pẹlu awọn eerun igi, isọpọ jinle sinu awọn olupin ifokansi fun awọn iwe aṣẹ gigun, ati awọn iyatọ aifwy fun fọnka tabi akiyesi window sisun. Gẹgẹbi awọn window ti o tọ titari si awọn miliọnu awọn ami-ami, awọn kernel ti o mọ IO bii eyi jẹ pataki si titọju ikẹkọ ati ṣiṣe awọn idiyele iṣakoso.

Real-World imuse

Ikẹkọ awọn awoṣe ede nla bii Llama ati awọn ọna ṣiṣe GPT ni iyara ati ni idiyele GPU kekere

Ṣiṣẹ awọn oluranlọwọ iwiregbe ọrọ-gigun ti o njẹ gbogbo awọn iwe tabi awọn koodu koodu laisi ṣiṣiṣẹ ti iranti

Iyara awọn opo gigun ti akopọ iwe-ipamọ ti o ṣe ilana awọn mewa ti ẹgbẹẹgbẹrun awọn ami ni ẹẹkan

Iran agbara ati awọn oluyipada multimodal nibiti awọn ilana gigun ti awọn abulẹ aworan jẹ ki akiyesi gbowolori

Awọn Ilana imuse

FlashAttention ni iwa

Ikẹkọ awọn awoṣe ede nla bii Llama ati awọn ọna ṣiṣe GPT ni iyara ati ni idiyele GPU kekere.

Ikẹkọ awọn awoṣe ede nla bii Llama ati awọn eto ara GPT ni iyara ati ni isalẹ idiyele GPU Awọn ẹgbẹ nigbagbogbo gba awọn abajade to dara julọ nigbati wọn ṣalaye awọn ilodi didara ni iwaju, tọju ọna imudara eniyan fun awọn ọran eti, ati tọpa awọn anfani iṣelọpọ mejeeji ati awọn idiyele aṣiṣe lori akoko.

FlashAttention ni iwa

Ṣiṣẹ awọn oluranlọwọ iwiregbe ọrọ-gigun ti o njẹ gbogbo awọn iwe tabi awọn koodu koodu laisi ṣiṣiṣẹ ti iranti.

Ṣiṣẹ awọn oluranlọwọ iwiregbe ọrọ-ọrọ gigun ti o jẹ gbogbo awọn iwe tabi awọn koodu koodu laisi ṣiṣiṣẹ ti iranti Awọn ẹgbẹ nigbagbogbo gba awọn abajade ti o dara julọ nigbati wọn ṣalaye awọn ilodi didara ni iwaju, tọju ọna igbega eniyan fun awọn ọran eti, ati tọpa mejeeji awọn anfani iṣelọpọ ati awọn idiyele aṣiṣe lori akoko.

FlashAttention ni iwa

Iyara awọn opo gigun ti akopọ iwe-ipamọ ti o ṣe ilana awọn mewa ti ẹgbẹẹgbẹrun awọn ami ni ẹẹkan.

Iyara awọn opo gigun ti akopọ iwe-ipamọ ti o ṣe ilana awọn mewa ti ẹgbẹẹgbẹrun awọn ami ni ẹẹkan Awọn ẹgbẹ nigbagbogbo gba awọn abajade to dara julọ nigbati wọn ṣalaye awọn ilodi didara ni iwaju, tọju ọna imudara eniyan fun awọn ọran eti, ati tọpa awọn anfani iṣelọpọ mejeeji ati awọn idiyele aṣiṣe lori akoko.

FlashAttention ni iwa

Iran agbara ati awọn oluyipada multimodal nibiti awọn ilana gigun ti awọn abulẹ aworan jẹ ki akiyesi gbowolori.

Iranran agbara ati awọn oluyipada multimodal nibiti awọn ilana gigun ti awọn abulẹ aworan jẹ ki akiyesi gbowolori Awọn ẹgbẹ nigbagbogbo gba awọn abajade to dara julọ nigbati wọn ṣalaye awọn ilodi didara ni iwaju, tọju ọna imudara eniyan fun awọn ọran eti, ati tọpa awọn anfani iṣelọpọ mejeeji ati awọn idiyele aṣiṣe lori akoko.

Awọn ewu & Awọn ọna iṣọ

!

Awọn otitọ ti a sọ di mimọ le tẹ awọn ijabọ sii ni idakẹjẹ, awọn ṣiṣan atilẹyin, tabi awọn abajade iwadii.

!

Ifamọ kiakia le ṣẹda awọn abajade aisedede kọja awọn ibeere ti o jọra.

!

Awọn data ọrọ ifarabalẹ le farahan ti awọn idari wiwọle ko lagbara.

Ilana Ilana imuse

1

Ṣetumo ọna kika iṣẹjade, ohun orin, ati awọn iṣedede didara ṣaaju ṣiṣejade.

Ṣetumo ọna kika iṣẹjade, ohun orin, ati awọn iṣedede didara ṣaaju ṣiṣejade. Ṣe itọju igbesẹ kọọkan bi ẹnu-ọna ẹri: ti awọn ibeere ko ba ni ibamu, daduro yiyọ kuro, pa aafo naa, ati lẹhinna faagun lilo.

2

Awọn idahun ilẹ pẹlu awọn orisun ti o gbẹkẹle nigbakugba ti deede ba ṣe pataki.

Awọn idahun ilẹ pẹlu awọn orisun ti o gbẹkẹle nigbakugba ti deede ba ṣe pataki. Ṣe itọju igbesẹ kọọkan bi ẹnu-ọna ẹri: ti awọn ibeere ko ba ni ibamu, daduro yiyọ kuro, pa aafo naa, ati lẹhinna faagun lilo.

3

Jeki aaye ayẹwo atunyẹwo eniyan fun awọn abajade ti o ga julọ.

Jeki aaye ayẹwo atunyẹwo eniyan fun awọn abajade ti o ga julọ. Ṣe itọju igbesẹ kọọkan bi ẹnu-ọna ẹri: ti awọn ibeere ko ba ni ibamu, daduro yiyọ kuro, pa aafo naa, ati lẹhinna faagun lilo.

4

Tọpinpin awọn ilana ikuna ati tunṣe awọn itọsi tabi ṣiṣan iṣẹ nigbagbogbo.

Tọpinpin awọn ilana ikuna ati tunṣe awọn itọsi tabi ṣiṣan iṣẹ nigbagbogbo. Ṣe itọju igbesẹ kọọkan bi ẹnu-ọna ẹri: ti awọn ibeere ko ba ni ibamu, daduro yiyọ kuro, pa aafo naa, ati lẹhinna faagun lilo.

Tesiwaju Ṣiṣawari