Skrive en GPU-kjernefunksjon
Uttrykk enhetskode i Mojo.
Skrive en GPU-kjernefunksjon er en gratis leksjon i Mojo Academy på CoddyKit. Dette er leksjon 3 av 4. Du kan lese valgfritt 3 leksjoner fra denne læringsstien gratis i sin helhet – deretter låser CoddyKit PRO opp alle leksjoner, samt praktisk øving med en innebygd kodeeditor og en AI-veileder som er tilgjengelig døgnet rundt. Den er en del av læringsløpet i Mojo Academy, og fremdriften din synkroniseres mellom nettet og CoddyKit-appen. Kurset i Mojo Academy inneholder totalt 4 leksjoner.
Hva en kjerne er
En GPU-kjerne er funksjonen hver tråd kjører. De skriver arbeidet for ett element, og maskinvaren gjentar det for mange elementer.
Tenk på én tråd
Trikset er å skrive en kjerne sett fra én enkelt tråds perspektiv. Hver tråd utfører sin egen lille del av arbeidet.
En kjerne er bare en fn
I Mojo er en kjerne en vanlig fn. Den strenge, typede egenskapen er nettopp det enhetskode trenger for å kompileres raskt.
fn add_kernel():
passSende inn buffere
En kjerne tar pekere til enhetsminne som parametere. Disse bufferne inneholder inndataene og utdataene den skal fylle.
fn add_kernel(a: UnsafePointer[Float32], out: UnsafePointer[Float32]):
passFinne denne trådens indeks
Inne i kjernen beregner hver tråd først sin egen posisjon. Denne indeksen velger hvilket element den skal behandle.
var i = block_idx.x * block_dim.x + thread_idx.xBeskytte grensene
Kontroller alltid indeksen før De berører minnet. En grensekontroll hindrer løse tråder i å lese forbi matrisen.
if i < n:
out[i] = a[i] + b[i]Utføre elementets arbeid
Selve kroppen er liten: les inndata, beregn, skriv ett resultat. Hele kjernen får ofte plass på én enkelt matematisk linje.
out[i] = a[i] * b[i]Starte kjernen
De starter den ved å velge et rutenett og en blokkstørrelse. Oppstarten sprer koden for én tråd over hele rutenettet.
ctx.enqueue_function[add_kernel](grid_dim=blocks, block_dim=256)Ingen returverdi
Kjerner returnerer ikke resultater til den som kaller dem. De skriver til utdatabufferen, som De leser tilbake etterpå.
Hold grenforgreninger på et minimum
Tråder fungerer best i låstakt. Omfattende grenforgrening får baner til å avvike og vente, så hold kjernelogikken enkel og ensartet.
Samme idé, enorm skala
Én kort kjerne pluss et stort rutenett gir millioner av resultater. Skalaen kommer fra oppstarten, ikke fra lengre kode.
Rask kontroll
De skriver kroppen til en GPU-kjerne for én tråd.
Oppsummering
En kjerne er en fn for én tråd: finn indeksen, kontroller grensene, utfør matematikken for ett element, og start deretter kjernen over et rutenett. ⚡
Lær deg Mojo med en AI-veileder – gratis
Skriv og kjør ekte kode i nettleseren, få umiddelbar hjelp fra en AI-veileder som er tilgjengelig døgnet rundt, og fortsett der du slapp – på nettet eller i appen.
- Kurs
- 30
- Leksjoner
- 120
Ofte stilte spørsmål
Er leksjonen «Skrive en GPU-kjernefunksjon» gratis?
Ja – du kan lese valgfritt 3 av leksjonene i læringsstien Mojo Academy, inkludert «Skrive en GPU-kjernefunksjon», gratis i sin helhet her på nettet. Deretter låser CoddyKit PRO opp alle leksjoner, samt interaktiv øving med en innebygd kodeeditor og en AI-veileder som er tilgjengelig døgnet rundt. Kurset i Mojo Academy inneholder totalt 4 leksjoner.
Hva lærer jeg i «Skrive en GPU-kjernefunksjon»?
Uttrykk enhetskode i Mojo. Du øver på Mojo Academy med praktisk kode som du kjører direkte i nettleseren, mens en AI-veileder som er tilgjengelig døgnet rundt, svarer på spørsmålene dine mens du jobber deg gjennom leksjonen.
Trenger jeg erfaring for å begynne med Mojo Academy?
Ingen tidligere erfaring er nødvendig. Mojo Academy på CoddyKit er lagt opp for både nybegynnere og viderekomne, så De kan begynne her eller helt fra start og lære i Deres eget tempo. Dette er leksjon 3 av 4.
Hvor lang tid tar leksjonen «Skrive en GPU-kjernefunksjon»?
De fleste CoddyKit-leksjoner tar omtrent 5–10 minutter. Hver leksjon er kort og interaktiv, slik at De gjør jevne fremskritt og kan fortsette akkurat der De slapp – både på nettet og i appen.
Kan jeg skrive og kjøre kode i denne Mojo Academy-leksjonen?
Ja. Alle Mojo Academy-leksjoner har en innebygd kodeeditor, slik at De kan skrive og kjøre ekte kode direkte i nettleseren og få umiddelbar tilbakemelding fra AI – uten lokal konfigurering.
Alle leksjonene i dette kurset
- Hvorfor GPU-er for AI-arbeidsbelastninger
- Tråder, blokker og rutenett
- Skrive en GPU-kjernefunksjon
- Flytte data til og fra enheten