使用cuda c ++查找数组中元素的索引

Question

我是 cuda 新手。

我有两个数组：

  int* AA = new int[5]{1,2,3,4,5};
  int* BB = new int[5]{ 2,2,2,4,4 };

我想找到 AA 中每个元素的索引，它等于 BB 中的每个元素，在这种情况下是

{1,1,1,3,3}

---

这是我的代码：

__global__ void findIndex(int* A, int* B, int* C)
{
    int i = threadIdx.x;


    for (int j = 0; j < 5; j++)
    {
        if (B[i] == A[j])
        {
            C[i] = j;

        }
    }
}


int main() {

    int* AA = new int[5]{1,2,3,4,5};
    int* BB = new int[5]{ 2,2,2,4,4 };
    int* CC = new int[5]{ 0,0,0,0,0 };

    int(*ppA), (*ppB), (*ppC);

    cudaMalloc((void**)&ppA, (5) * sizeof(int));
    cudaMalloc((void**)&ppB, (5) * sizeof(int));
    cudaMalloc((void**)&ppC, (5) * sizeof(int));

    cudaMemcpy(ppA, AA, 5 * sizeof(int), cudaMemcpyHostToDevice);
    cudaMemcpy(ppB, BB, 5 * sizeof(int), cudaMemcpyHostToDevice);
    cudaMemcpy(ppC, CC, 5 * sizeof(int), cudaMemcpyHostToDevice);

    int numBlocks = 1;
    dim3 threadsPerBlock(5);
    findIndex << <numBlocks, threadsPerBlock >> > (ppA, ppB, ppC);


    cudaMemcpy(CC, ppC, 5 * sizeof(int), cudaMemcpyDeviceToHost);
    for (int m = 0; m < 5; m++) {
        printf("%d ", CC[m]);
    }
}

我的输出是：

{1,2,3,0,0}

谁能帮忙？

Answer 1

最简单的非稳定单 GPU 解决方案是使用原子，如下所示：

__global__ void find(int * arr,int * counter, int * result)
{
    int id = blockIdx.x*blockDim.x+threadIdx.x;
    if(arr[id] == 4)
    {
        int ctr = atomicAdd(counter,1);
        result[ctr] = id;
    }
}

通过这种方式，您可以在“结果”数组中获得一个结果数组，如果想要的数字很少，它不会减慢很多（就像整个源数组中只有少数）。不过，这不是多 GPU 系统的最佳方式。需要 gpus 之间的主机端协调，除非使用最新工具包中的特殊 CUDA 功能（用于系统级原子）。

如果 4 的数量导致“密集”arr 数组或者如果您有多个 GPU，那么您应该寻找其他解决方案，例如流压缩。首先选择包含 4 的单元格作为掩码。然后进行压实。一些 Nvidia 博客或教程有此算法。

对于“原子”解决方案（尤其是“共享”内存原子），Maxwell（及更高版本）架构比 Kepler 要好得多，以防万一您仍然使用 Kepler。由于无法知道原子操作的顺序，因此使用原子也不能完全重现。 大多数时候你会得到一个不同顺序的结果数组。但是流压缩保留了结果顺序。这可能使您免于在其之上编写排序算法（如双调排序、剪切排序等）。