arrays - 如何从子数组大小并行计算数组的偏移量？

Question

我有一个数据结构数组，在数组中按不同大小的块排序。算法从中提取少量数据结构，并将其存储在一个新结构中。它还计算从每个块中取出的数量。然而，计算着色器的近乎不同步特性导致结果数组未排序。我想制作一个新的，它会使用新块大小的知识（算法过去从每个块中提取了多少）来对它们进行排序。但是，虽然我知道块的大小，但我不知道数组中的偏移量，每个块应该从哪里开始。我可以在 CPU 上轻松完成，但一切都发生在 GPU 内——它是图形输入的预处理，数量巨大，来回通信，在帧中间等待 CPU，成本很高。

如何在 GPU 着色器中有效地计算所述偏移量？这对我来说似乎是一个并行减少问题，但是需要保存中间结果。

标签： arraysparallel-processinggpgpu

Answer 1

对于 NVIDIA GPU，您可以在推力和 CUB 中找到并行“前缀和”的库和代码。如果您将块大小存储在一个数组中，那么该数组上的前缀总和应该为您提供每个块的偏移量。

幼崽：https ://nvlabs.github.io/cub/structcub_1_1_device_scan.html

推力：https ://docs.nvidia.com/cuda/thrust/index.html#prefix-sums