はてなブックマーク

NVIDIAの開発系BlogになるほどなーなTips: Grid-stride loop を見つけたので軽く解説。 (元ネタはコチラ) 毎度おなじみ SAXPY : ベクトルの積和演算 Y[i] = a * X[i] + Y[i] (i = 0,1 ... n-1) これをCPUでフツーに行うとき、iについてくるくる回すloopで実装しますわな。 void saxpy(int n, float a, const float* x, float* y) { for ( int i = 0; i < n; ++i ) { y[i] = a * x[i] + y[i]; } } かたやこいつをGPU(CUDA)でやるときの常套手段は "iによるn回のloop" を "n個のスレッド"　に置き換えることで コードからloopを取っ払います。 __global__ void kernel_saxpy