threadIdx, blockIdx, blockDim

Każdy wątek wykonuje ten sam kod, więc każdy musi zadać pytanie: który element danych należy do mnie? CUDA odpowiada za pomocą wbudowanych zmiennych. 📍

Wewnątrz kernela threadIdx informuje wątek o jego pozycji w należącym do niego bloku. Pierwszy wątek w bloku ma wartość threadIdx.x równą 0.

Zmienna blockIdx informuje wątek, do którego bloku należy on w obrębie siatki. Numerowanie bloków również zaczyna się od 0.

Otrzymujesz także blockDim — liczbę wątków w bloku. Jeśli uruchomiono 256 wątków, wartość blockDim.x wynosi 256 dla każdego wątku.

Nigdy nie ustawiasz tych zmiennych samodzielnie. CUDA wypełnia je dla każdego wątku w chwili uruchomienia kernela.

Każda z tych zmiennych ma pola x, y i z dla uruchomień 1D, 2D lub 3D. W przypadku prostych tablic zwykle używa się tylko pola x.

Samodzielnie threadIdx powtarza się w każdym bloku. Aby uzyskać unikalną pozycję, musisz połączyć blockIdx i blockDim z threadIdx.

Przy blockDim równym 4 wątek 2 w bloku 0 odpowiada elementowi 2, natomiast wątek 2 w bloku 1 — elementowi 6. Blok przesuwa punkt początkowy.

Te wbudowane elementy są wewnątrz kernela tylko do odczytu. Traktuj je jak zaufane współrzędne dokładnie opisujące położenie Twojego wątku.

Jest jeszcze jeden pomocny element: gridDim przechowuje łączną liczbę bloków w siatce. Przydaje się, gdy iterujesz po większej liczbie danych niż liczba wątków.

Te trzy zmienne są podstawą obliczania indeksu globalnego, dzięki któremu każdy wątek może przypisać sobie własny element.

Poznałeś wbudowane elementy: threadIdx określa pozycję w bloku, blockIdx — blok, a blockDim — rozmiar bloku. Razem precyzyjnie wskazują konkretny wątek. ✅