您必须分配一个 OpenCL 缓冲区并将您的 CPU 数据复制到其中。OpenCL 缓冲区具有固定大小,因此如果数据大小发生变化,您要么必须重新创建它,要么将其设置为“足够大”,如果需要更少的内存,则只使用它的一小部分。例如,要为设备创建缓冲区b
并同时将其所有数据复制到设备:
cl_mem buffer_b = clCreateBuffer(
context, // OpenCL context
CL_MEM_READ_ONLY | CL_MEM_COPY_HOST_PTR, // Only read access from kernel,
// copy data from host
sizeof(cl_double) * b.size(), // Buffer size in bytes
&b[0], // Pointer to data to copy
&errorcode); // Return code
也可以直接映射主机内存(CL_MEM_USE_HOST_PTR
),但这会对对齐和创建缓冲区后对主机内存的访问施加一些限制。基本上,当您当前未映射主机内存时,它可能包含垃圾。
这取决于。第二维向量的大小是否始终相等?然后在将它们上传到 OpenCL 设备时将它们展平。否则会变得更复杂。
您将缓冲区参数声明为__global
内核中的指针。例如,__global double *b
将适用于在 1 中创建的缓冲区。您可以简单地在内核中使用数组表示法来访问缓冲区中的各个元素。
您无法从内核中查询缓冲区大小,因此您必须手动传递它。这也可以隐式发生,例如,如果工作项的数量与b
.
可以访问所有计算数据的内核可能如下所示:
__kernel void foo(long x, double y, double a, __global double* b, int b_size,
__global long* c, __global double* d,
__global double* result) {
// Here be dragons
*result = 0.0;
}
请注意,您还必须为结果分配内存。如果需要,可能需要传递额外的大小参数。您可以按如下方式调用内核:
// Create/fill buffers
// ...
// Set arguments
clSetKernelArg(kernel, 0, sizeof(cl_long), &x);
clSetKernelArg(kernel, 1, sizeof(cl_double), &y);
clSetKernelArg(kernel, 2, sizeof(cl_double), &a);
clSetKernelArg(kernel, 3, sizeof(cl_mem), &b_buffer);
cl_int b_size = b.size();
clSetKernelArg(kernel, 4, sizeof(cl_int), &b_size);
clSetKernelArg(kernel, 5, sizeof(cl_mem), &c_buffer);
clSetKernelArg(kernel, 6, sizeof(cl_mem), &d_buffer);
clSetKernelArg(kernel, 7, sizeof(cl_mem), &result_buffer);
// Enqueue kernel
clEnqueueNDRangeKernel(queue, kernel, /* ... depends on your domain */);
// Read back result
cl_double result;
clEnqueueReadBuffer(queue, result_buffer, CL_TRUE, 0, sizeof(cl_double), &result,
0, NULL, NULL);