我为openCL修改了一系列浮点数的并行缩减示例。现在我想扩展代码以包含cl_float3。所以我想在cl_float3数组中找到最小值。我认为这是从内核到float3的直接扩展。但是当我从内核返回时,我正在接收垃圾值。以下是内核:
__kernel void pmin3(__global float3 *src,
__global float3 *gmin,
__local float3 *lmin,
__global float *dbg,
uint nitems,
uint dev)
{
uint count = nitems / get_global_size(0);
uint idx = (dev == 0) ? get_global_id(0) * count
: get_global_id(0);
uint stride = (dev == 0) ? 1 : get_global_size(0);
// Private min for the work-item
float3 pmin = (float3)(pow(2.0,32.0)-1,pow(2.0,32.0)-1,pow(2.0,32.0)-1);
for (int n = 0; n < count; n++, idx += stride) {
pmin.x = min(pmin.x,src[idx].x);
pmin.y = min(pmin.y,src[idx].y);
pmin.z = min(pmin.z,src[idx].z);
}
// Reduce values within the work-group into local memory
barrier(CLK_LOCAL_MEM_FENCE);
if (get_local_id(0) == 0)
lmin[0] = (float3)(pow(2.0,32.0)-1,pow(2.0,32.0)-1,pow(2.0,32.0)-1);
for (int n = 0; n < get_local_size(0); n++) {
barrier(CLK_LOCAL_MEM_FENCE);
if (get_local_id(0) == n) {
lmin[0].x = min(lmin[0].x,pmin.x);
lmin[0].y = min(lmin[0].y,pmin.y);
lmin[0].z = min(lmin[0].z,pmin.z);
}
}
barrier(CLK_LOCAL_MEM_FENCE);
// Write to __global gmin which will contain the work-group minima
if (get_local_id(0) == 0)
gmin[get_group_id(0)] = lmin[0];
// Collect debug information
if (get_global_id(0) == 0) {
dbg[0] = get_num_groups(0);
dbg[1] = get_global_size(0);
dbg[2] = count;
dbg[3] = stride;
}
}
__kernel void min_reduce3( __global float3 *gmin)
{
for (int n = 0; n < get_global_size(0); n++) {
barrier(CLK_GLOBAL_MEM_FENCE);
if (get_global_id(0) == n) {
gmin[0].x = min(gmin[0].x,gmin[n].x);
gmin[0].y = min(gmin[0].y,gmin[n].y);
gmin[0].z = min(gmin[0].z,gmin[n].z);
}
}
barrier(CLK_GLOBAL_MEM_FENCE);
}
我认为这是get_global_id(0)和get_global_size()的问题,它给出了整个大小而不是唯一给出的行数。有什么建议吗?
答案 0 :(得分:3)
正如其他人所提到的,float3
(和其他type3
类型)表现为 float4 (和其他type4
类型),用于大小和对齐。使用内置的vec_step
函数也可以看到这一点,该函数返回输入对象类型中的元素数,但对type3
个对象返回4。
如果您的主机代码生成压缩 float3
数组 - 每个对象只占用3个浮点数的大小和对齐 - 那么从OpenCL使用它的正确方法是:
float*
参数代替float3*
vload3
vstore3
答案 1 :(得分:1)
float3 是16字节对齐的。见OpenCL specs 6.1.5。