如何在OpenCL中对全局计数器进行原子递增

时间:2016-08-12 16:52:46

标签: opencl global atomic pyopencl

我希望OpenCL中有一个全局计数器,可以通过每个工作组中的每个工作项来增加。

在我的内核中我做了:

#pragma OPENCL EXTENSION cl_khr_global_int32_base_atomics : enable

void increase(volatile __global int* counter)
{
    atomic_inc(counter);
}

__kernel void test()
{  
    volatile __global int* counter = 0; 
    increase(counter);
    printf("Counter: %i",&counter);
}

我的Hostcode是将内核排入队列的最小pyopencl:

import pyopencl as cl

platform = cl.get_platforms()[0]
devs = platform.get_devices()
device = devs[0]
ctx = cl.Context([device])
queue = cl.CommandQueue(ctx)
mf = cl.mem_flags

f = open('Minimal.cl', 'r')
fstr = "".join(f.readlines())
prg = cl.Program(ctx, fstr).build()
test_knl = prg.test

def f():
     cl.enqueue_nd_range_kernel(queue,test_knl,(1,1,2),None)
f()

我希望输出显示(可能是随机排列的)"Counter: i"的外观,其中i是总工作项的数量(在我的情况下为2)。 相反,我没有得到打印输出。当我重新运行程序时,它失败了

pyopencl.cffi_cl.LogicError: clcreatecontext failed: <unknown error -9999>

完全杀死我的IDE(Spyder)。

1 个答案:

答案 0 :(得分:4)

volatile __global int* counter = 0; 

创建指向全局内存的指针是不够的。它背后必须有一些全局存储器。

至少有两种选择:

1)如果使用OpenCL 2.0实现,则可以创建程序范围变量:

void increase(volatile __global int* counter)
{
  atomic_inc(counter);
}

__global int counter = 0;

__kernel void test()
{  
  volatile __global int* counterPtr = &counter; 
  increase(counterPtr); // or increase(&counter);
  printf("Counter: %i",*counterPtr);
}

2)创建一个OpenCL缓冲区并通过内核参数传递它:

void increase(volatile __global int* counter)
{
  atomic_inc(counter);
}

__kernel void test(__global int *counterArg)
{  
  volatile __global int* counterPtr = counterArg; 
  increase(counterPtr); // or increase(counterArg);
  printf("Counter: %i",*counterPtr);
}