Mojo Academy · Lektion

Optimér det indre produkt

Vektorisér og udfold prikproduktet.

Lektion 3 af 413 trin

Optimér det indre produkt er en gratis Mojo Academy-lektion på CoddyKit. Dette er lektion 3 af 4. Du kan læse alle 3 lektioner i dette læringsspor gratis i deres fulde længde — derefter låser CoddyKit PRO alle lektioner op samt praktiske øvelser med en indbygget kodeeditor og en AI-underviser døgnet rundt. Den er en del af læringsforløbet i Mojo Academy, og dine fremskridt synkroniseres på tværs af nettet og CoddyKit-appen. Mojo Academy-kurset indeholder 4 lektioner i alt.

Prikproduktets kerne

Det indre produkt multiplicerer to sekvenser element for element og lægger resultaterne sammen. Hvis du gør det hurtigere, gør du hele din matmul hurtigere.

var acc: Float32 = 0.0
for k in range(K):
    acc += a[k] * b[k]

Én lane ad gangen er langsomt

En almindelig løkke håndterer ét par pr. trin. Moderne CPU'er kan håndtere mange på én gang, så skalar kode efterlader lanes inaktive.

Introduktion til SIMD

En SIMD-værdi pakker flere tal og anvender én operation på dem alle samtidig, så der udføres reelt arbejde parallelt pr. instruktion.

var v = SIMD[DType.float32, 4](1, 2, 3, 4)

Indlæs en blok på én gang

I stedet for ét flydende tal kan du indlæse et helt udsnit af hvert array med SIMD-bredde i én enkelt flytning, klar til vektorberegning.

var va = a.load[width=4](k)
var vb = b.load[width=4](k)

Multiplicer hele vektorer

Multiplikation af to SIMD-værdier giver en vektor med produkter i ét trin. Det elementvise arbejde foregår på alle lanes samtidig.

var prod = va * vb  # 4 products at once

Akkumuler i en vektor

Bevar en SIMD-akkumulator, og læg hver produktvektor ind i den. Du venter med den endelige sum, til løkken er færdig.

var acc = SIMD[DType.float32, 4](0)
acc += va * vb

Gå frem med bredden

Løkken går nu frem med SIMD-bredden i stedet for én. En bredde på fire lanes betyder en fjerdedel så mange gentagelser.

for k in range(0, K, 4):
    acc += a.load[width=4](k) * b.load[width=4](k)

Reducer til sidst

Efter løkken skal du samle lane-akkumulatoren til ét tal med en vandret reduktion, som giver det endelige prikprodukt.

var total = acc.reduce_add()

Udfold løkken

Udfoldning behandler flere bredder pr. gentagelse, reducerer løkkens overhead og gør mere uafhængigt arbejde synligt for CPU'en.

Håndter resten

Hvis K ikke er deleligt med bredden, er der nogle elementer tilbage. En lille hale-løkke færdiggør dem ét ad gangen.

for k in range(K - K % 4, K):
    total += a[k] * b[k]

Lad Mojo hjælpe

Mojo tilbyder hjælpefunktionen vectorize, som anvender en breddestyret krop på et interval og håndterer trin og hale for dig.

Hurtigt tjek

Hvorfor kalder du reduce_add til sidst efter en prikproduktløkke med SIMD-bredde?

Besvar den i den interaktive lektion

Opsummering

Vektorisér det indre produkt ved at indlæse SIMD-blokke, multiplicere lanes med hinanden og akkumulere, hvorefter du reducerer; udfold løkken, og håndtér halen. ⚡

Gratis at komme i gang

Lær Mojo med en AI-underviser — gratis

Skriv og kør rigtig kode i din browser, få øjeblikkelig hjælp fra en AI-underviser døgnet rundt, og fortsæt, hvor du slap, på web eller i appen.

Kurser: 30
Lektioner: 120

Start gratis Se hele kurset

Ofte stillede spørgsmål

Er lektionen “Optimér det indre produkt” gratis?

Ja — alle 3 lektioner i læringssporet Mojo Academy, inklusive “Optimér det indre produkt”, kan læses gratis i deres fulde længde her på webstedet. Derefter låser CoddyKit PRO alle lektioner op samt interaktive øvelser med en indbygget kodeeditor og en AI-underviser døgnet rundt. Mojo Academy-kurset indeholder 4 lektioner i alt.

Hvad lærer jeg i “Optimér det indre produkt”?

Vektorisér og udfold prikproduktet. Du øver dig i Mojo Academy med praktisk kode, som du kører direkte i browseren, og en AI-vejleder døgnet rundt besvarer dine spørgsmål, mens du arbejder dig gennem lektionen.

Skal jeg have erfaring for at begynde på Mojo Academy?

Der kræves ingen tidligere erfaring. Mojo Academy på CoddyKit er tilrettelagt for både begyndere og øvede, så du kan starte her eller fra begyndelsen og lære i dit eget tempo. Dette er lektion 3 af 4.

Hvor lang tid tager lektionen “Optimér det indre produkt”?

De fleste CoddyKit-lektioner tager cirka 5–10 minutter. Hver lektion er kort og interaktiv, så du gør løbende fremskridt og kan fortsætte, hvor du slap – på både web og app.

Kan jeg skrive og køre kode i denne Mojo Academy-lektion?

Ja. Alle Mojo Academy-lektioner har en indbygget kodeeditor, så du kan skrive og køre rigtig kode direkte i din browser og få øjeblikkelig feedback fra AI – uden lokal opsætning.

Alle lektioner i dette kursus

← Tilbage til Mojo Academy