使用 OpenCL 对 GPU 缓冲区进行 memset() 的最快方法是什么？

Question

我正在使用 OpenCL，我需要 memset() 全局设备内存中的某个数组。 CUDA 有一个类似memset() 的 API 函数，但 OpenCL 没有。我阅读了this，在那里我找到了两种可能的选择：

1. 在主机上使用 memset() 和一些暂存缓冲区，然后 clEnqueueWriteBuffer() 将其复制到设备上的缓冲区。

2. 将以下内核入队：

   __kernel void memset_uint4(
       __global  uint4* mem,
       __private uint4  val) 
   {
       mem[get_global_id(0)] = val; 
   }
   

哪个更好？或者更确切地说，在哪种情况下/哪个平台比另一个更好？

注意：如果归零内存的特殊情况值得特殊处理，那也很高兴知道。

Answer 1

您可以使用 OpenCL v1.2 中的clEnqueueFillBuffer()。这正是你所需要的。并且在如何用模式填充缓冲区方面非常灵活。

如果您使用的是 1.1 或更低版本....那么您必须求助于其他方法。

Answer 2

快速完成此操作的一个好方法（如果您有额外的可用内存）是拥有一个预先确定大小的初始化数组（例如一个全零填充的数组），然后在需要归零时随时在设备上复制出缓冲区。以我的经验，这比填写 OpenCL 或 CUDA 的任何调用都要快得多。显然这是一种特殊情况，但在我上次测试时要快得多。