CUDA的__shared__内存何时有用?

时间:2011-11-04 15:07:45

标签: c cuda gpu

有人可以帮我一个关于如何使用共享内存的简单示例吗? Cuda C编程指南中包含的示例似乎与不相关的细节混杂在一起。

例如,如果我将一个大型数组复制到设备全局内存并想要对每个元素求平方,那么如何使用共享内存来加快速度呢?或者在这种情况下没用?

2 个答案:

答案 0 :(得分:27)

在你提到的特定情况下,共享内存没用,原因如下:每个数据元素只使用一次。要使共享内存有用,您必须使用良好的访问模式多次使用传输到共享内存的数据来提供帮助。原因很简单:只需从全局内存中读取就需要1个全局内存读取和零共享内存读取;首先将它读入共享内存需要1次全局内存读取和1次共享内存读取,这需要更长的时间。

这是一个简单的例子,其中块中的每个线程计算相应的值,平方,加上其左右邻居的平均值,平方:

  __global__ void compute_it(float *data)
  {
     int tid = threadIdx.x;
     __shared__ float myblock[1024];
     float tmp;

     // load the thread's data element into shared memory
     myblock[tid] = data[tid];

     // ensure that all threads have loaded their values into
     // shared memory; otherwise, one thread might be computing
     // on unitialized data.
     __syncthreads();

     // compute the average of this thread's left and right neighbors
     tmp = (myblock[tid > 0 ? tid - 1 : 1023] + myblock[tid < 1023 ? tid + 1 : 0]) * 0.5f;
     // square the previousr result and add my value, squared
     tmp = tmp*tmp + myblock[tid] * myblock[tid];

     // write the result back to global memory
     data[tid] = myblock[tid];
  }

请注意,这可以设想只使用一个块。更多块的扩展应该是直截了当的。假设块尺寸(1024,1,1)和网格尺寸(1,1,1)。

答案 1 :(得分:9)

将共享内存视为显式托管缓存 - 只有在需要多次访问数据时才会有用,无论是在同一个线程内还是在同一个块中的不同线程中< / em>的。如果您只访问一次数据,则共享内存不会对您有所帮助。