1

我需要使用归约来计算 N 个信号的平均值。输入是一个大小为 MN 的一维数组，其中 M 是每个信号的长度。

最初我有额外的共享内存来首先复制数据并减少每个信号。但是，原始数据已损坏。

我的程序试图最小化共享内存。所以我想知道如何使用寄存器对 N 个信号进行归约和。我有 N 个线程，一个共享内存 (float) s_m[N*M], 0....M-1 是第一个信号，等等。

我是否需要 N 个寄存器（或一个）来存储 N 个不同信号的平均值？（我知道如何使用多线程编程和 1 个寄存器进行顺序加法）。我想做的下一步是从对应信号的平均值中减去输入中的每个值。

4

2 回答 2

3

您的问题非常小（N = 32 且 M < 128）。但是，一些准则：

假设您要减少 N 个线程中每个线程的 N 个值。

  • 如果 N 非常大（> 成千上万），则只需在每个线程中按顺序对 M 进行缩减。
  • 如果 N 小于千分之十，请考虑使用一个 warp 或一个线程块来执行 N 个缩减中的每一个。
  • 如果 N 非常小但 M 非常大，请考虑在每次 N 减少时使用多个线程块。
  • 如果 N 非常小且 M 非常小（如您的数字），则仅在生成和/或消耗缩减的输入/输出的计算也在 GPU 上运行时才考虑使用 GPU 进行缩减。
于 2012-09-10T06:24:35.217 回答
-1

根据我对这个问题的理解，我说你不需要 N 个寄存器来存储 N 个不同信号的平均值。

如果您已经有 N 个线程 [假设每个线程只对一个信号进行归约]，那么您不需要 N 个寄存器来存储一个信号的归约。您只需要一个寄存器来存储平均值。

dim3 threads (N,1);
reduction<<<threads,1>>>(signals);  // signals is the [N*M] array 

__global__ reduction (int *signals)
{
   int id = threadIdx.x;
   float meanValue = 0.0;

   for(int i = 0; i < M; i++)
          meanValue = signals[id*M +i];

   meanValue =  meanValue/M;

   // Then do the subtraction
   for(int i = 0; i < M; i++)
          signals[id*M +i] -= meanValue;
}

如果你需要对N个不同信号的所有meanValues进行全局归约，那么你需要使用2个寄存器[一个存储局部均值，另一个存储全局均值]和共享内存

dim3 threads (N,1);
reduction<<<threads,1>>>(signals);  // signals is the [N*M] array 

__global__ reduction (int *signals)
{
   __shared__ float means[N];      // shared value
   int id = threadIdx.x;
   float meanValue = 0.0;
   float globalMean = 0.0;

   for(int i = 0; i < M; i++)
          meanValue += signals[id*M +i];

   means[id] =  meanValue/M;

   __syncthreads();

   // do the global reduction
   for(int i = 0; i < N; i++)
          globalMean += means[i];

   globalMean = globalMean/N;

   // Then do the subtraction
   for(int i = 0; i < M; i++)
          signals[id*M +i] -= globalMean;
}

我希望这可以帮助你。有任何疑问，请告诉我。

于 2011-07-24T05:26:36.947 回答