기본 인덱스 공식

CUDA 커널의 핵심은 모든 스레드가 데이터 조각 하나를 처리하는 것입니다. 그러려면 각 스레드에 고유한 전역 인덱스가 필요합니다.

블록 내부에서 threadIdx.x는 0부터 blockDim에서 1을 뺀 값까지만 셉니다. 여러 블록이 같은 작은 숫자를 재사용하므로 최종 인덱스로 사용할 수 없습니다.

그리드를 끝과 끝이 맞닿은 블록들로 이루어진 것으로 생각해 보십시오. 각 블록은 배열의 연속된 구간을 담당하며, blockIdx.x가 현재 어느 구간에 있는지 알려 줍니다.

blockDim.x는 블록당 스레드 수입니다. 각 구간의 너비이므로 블록 오프셋을 올바른 위치로 확장하는 데 사용됩니다.

세 값을 결합하십시오. 앞선 블록들을 건너뛴 다음 현재 블록 안에서 자신의 위치를 더합니다. 이 한 줄로 모든 스레드에 고유한 인덱스가 부여됩니다.

블록당 스레드가 4개라면 블록 0은 0부터 3, 블록 1은 4부터 7, 블록 2는 8부터 11을 담당합니다. 오프셋은 서로 겹치지 않습니다.

blockDim이 256일 때 블록 3의 스레드 2는 3 곱하기 256 더하기 2인 770에 도달합니다. 이것이 데이터에서 해당 스레드의 전역 위치입니다.

i를 얻은 뒤에는 이를 배열 첨자로 취급합니다. 각 스레드는 겹치지 않도록 자신의 요소만 읽고 씁니다.

n개의 스레드를 실행하면 공식은 0부터 n에서 1을 뺀 값까지의 모든 값을 정확히 한 번씩 만듭니다. 배열 전체를 완벽하게 일대일로 덮는 것입니다.

항상 더하기 전에 곱하십시오. blockIdx.x * blockDim.x는 구간의 시작점이고, threadIdx.x는 그 안에서의 위치입니다.

같은 개념을 .y 및 .z 멤버를 사용해 2차원과 3차원으로 확장할 수 있지만, 평탄한 배열에서는 .x 공식만 있으면 됩니다. 🚀

모든 커널에서 사용하는 공식을 배웠습니다. blockIdx.x * blockDim.x + threadIdx.x는 각 스레드에 배열 안의 고유한 슬롯 하나를 할당합니다. 🎉