Pisanie funkcji jądra GPU
Wyrażaj kod urządzenia w Mojo.
Pisanie funkcji jądra GPU to bezpłatna lekcja Mojo Academy na CoddyKit. To lekcja 3 z 4. Możesz przeczytać całą lekcję poniżej za darmo — a potem ćwiczyć ją interaktywnie w przeglądarce z wbudowanym edytorem kodu i tutorem AI dostępnym 24/7. To część ścieżki edukacyjnej Mojo Academy, a Twój postęp synchronizuje się między webem a aplikacją CoddyKit. Kurs Mojo Academy zawiera 4 lekcji w sumie.
Czym jest kernel
GPU kernel to funkcja wykonywana przez każdy wątek. Określasz pracę dla jednego elementu, a sprzęt powtarza ją dla wielu elementów.
Myśl z perspektywy jednego wątku
Kluczem jest napisanie kernela z perspektywy pojedynczego wątku. Każdy wątek wykonuje własny niewielki fragment pracy.
Kernel to po prostu fn
W Mojo kernel jest zwykłą funkcją fn. Jego ścisła, typowana natura jest dokładnie tym, czego kod urządzenia potrzebuje do szybkiej kompilacji.
fn add_kernel():
passPrzekazywanie buforów
Kernel przyjmuje jako parametry wskaźniki do pamięci urządzenia. Te bufory przechowują dane wejściowe oraz wynik, który kernel zapisze.
fn add_kernel(a: UnsafePointer[Float32], out: UnsafePointer[Float32]):
passZnajdowanie indeksu tego wątku
Wewnątrz kernela każdy wątek najpierw oblicza własną pozycję. Ten indeks wskazuje element, który wątek musi przetworzyć.
var i = block_idx.x * block_dim.x + thread_idx.xSprawdzanie granic
Zawsze sprawdzaj indeks przed dostępem do pamięci. Test zakresu chroni przed odczytem poza końcem tablicy przez niepotrzebne wątki.
if i < n:
out[i] = a[i] + b[i]Wykonywanie pracy dla elementu
Treść jest niewielka: odczytaj dane wejściowe, wykonaj obliczenia i zapisz jeden wynik. Cały kernel często mieści się w jednym matematycznym wierszu.
out[i] = a[i] * b[i]Uruchamianie kernela
Uruchamiasz go, wybierając rozmiar siatki i bloku. Uruchomienie rozpowszechnia kod przeznaczony dla jednego wątku na całą siatkę.
ctx.enqueue_function[add_kernel](grid_dim=blocks, block_dim=256)Brak wartości zwracanej
Kernele nie zwracają wyników do wywołującego. Zapisują je w buforze wyjściowym, który odczytujesz później.
Ogranicz rozgałęzienia
Wątki najlepiej działają synchronicznie. Rozbudowane rozgałęzienia powodują rozbieżność ścieżek i oczekiwanie części wątków, dlatego logika kernela powinna być prosta i jednolita.
Ta sama idea w ogromnej skali
Jeden krótki kernel oraz duża siatka dają miliony wyników. Skala wynika z uruchomienia, a nie z dłuższego kodu.
Szybki test
Piszesz treść kernela GPU przeznaczoną dla jednego wątku.
Podsumowanie
Kernel to funkcja fn dla jednego wątku: znajdź swój indeks, sprawdź granice, wykonaj obliczenia dla jednego elementu, a następnie uruchom kernel na siatce. ⚡
Ucz się Mojo dzięki korepetycjom AI — za darmo
Pisz i uruchamiaj kod w przeglądarce, otrzymuj natychmiastową pomoc od korepetytora AI dostępnego 24/7 i kontynuuj naukę w sieci lub w aplikacji.
- Kursy
- 30
- Lekcje
- 120
Często zadawane pytania
Czy lekcja „Pisanie funkcji jądra GPU” jest bezpłatna?
Tak — pełny tekst „Pisanie funkcji jądra GPU” jest dostępny za darmo tutaj w sieci. Aby ćwiczyć ją interaktywnie (wbudowany edytor kodu i tutor AI dostępny 24/7) i odblokować resztę kursu Mojo Academy, przejdź na CoddyKit PRO. Kurs Mojo Academy zawiera 4 lekcji w sumie.
Co nauczysz się w „Pisanie funkcji jądra GPU”?
Wyrażaj kod urządzenia w Mojo. Ćwiczysz Mojo Academy z praktycznym kodem, który uruchamiasz bezpośrednio w przeglądarce, a tutor AI dostępny 24/7 odpowiada na Twoje pytania podczas pracy nad lekcją.
Czy potrzebuję doświadczenia, aby zacząć Mojo Academy?
Nie wymagamy żadnego doświadczenia. Mojo Academy w CoddyKit jest strukturyzowany dla początkujących i zaawansowanych użytkowników, więc możesz zacząć tutaj lub od początku i uczyć się w swoim tempie. To lekcja 3 z 4.
Ile czasu zajmuje lekcja „Pisanie funkcji jądra GPU”?
Większość lekcji CoddyKit trwa około 5–10 minut. Każda lekcja to mały, interaktywny krok, dzięki czemu robisz systematyczne postępy i zawsze wracasz dokładnie do tego samego miejsca — na webie i w aplikacji.
Czy mogę pisać i uruchamiać kod w tej lekcji Mojo Academy?
Tak. Każda lekcja Mojo Academy zawiera wbudowany edytor kodu, więc piszesz i uruchamiasz prawdziwy kod bezpośrednio w przeglądarce i od razu otrzymujesz sprzężenie zwrotne od AI — bez konfiguracji na komputerze.
Wszystkie lekcje w tym kursie
- Dlaczego GPU do obciążeń AI
- Wątki, bloki i siatki
- Pisanie funkcji jądra GPU
- Przenoszenie danych na urządzenie i z urządzenia