python - 在 C/openCL 中寻址 vector 元素

Question

我正在 pyopenCL 中编写一个 openCL 内核，我想在其中处理 vector 元素。

在纯 C 中，我想要的结果是:

int i = 0;
float *vec = (float*)maalloc(sizeof(float)*4);
for (i=0;i<4;i++)
{
    vec[i]=2*i;
}

在 openCL 中， vector 的元素以“pythonic”点语法风格访问。

float4 vec = (float4)(0);
for (i=0;i<4,i++)
{
    vec.si = 2*i;
/*obviously doesn't work*/
}

所以 vec[2] 在 openCL 中变成了 vec.s2 ，所以用变量访问元素不再是直截了当的。尽管如此，如何使用变量访问 vector 元素？

Answer 1

OpenCL 支持主机端组件和设备端组件(内核)的 C，因此您可以使用 float 数组编写与第一个示例几乎完全相同的内核。内核可能如下所示:

__kernel void vectorAddition(__global float* vec) {
  // Get the global thread id in x dimension(eliminates loop)
  size_t index = get_global_id(0);

  vec[index] = 2.0f * index;
}

然后您可以指定要使用的线程数，以便对数组的每个元素执行此操作(使线程数与数组中的元素数相同)。

OpenCL 确实允许使用点符号进行访问，但这是为了访问 vector data types 的元素。 . vector 数据类型可以提供改进的性能，因为可以同时对 vector 数据类型中的所有元素执行相同的操作。

例如，float4 是一种 vector 数据类型，它存储四个彼此相邻的 32 位 float 以构成一个 128 位结构。然后，您可以一次对所有 4 个 float 执行操作。

例如:

float4 v = (float4)(1.0f, 2.0f, 3.0f, 4.0f);
float4 mult_result = v * 2;

这需要一条指令同时进行四个乘法。 mult_result 的值为 {2.0f, 4.0f, 6.0f, 8.0f}。

然后可以使用点符号来访问 float4 变量的组件，例如:

float a = v.x;           // a = 1.0f
float b = mult_result.y; // b = 4.0f

这里是 vector 数据类型的总结:Vector Data Type Overview .