opencl程序架构

来源:互联网 发布:淘宝网创始人有多少人 编辑:程序博客网 时间:2024/05/16 18:09

这几天研究了一下OpenCL,阅读了nvidia中的例子,归纳OpenCL程序的一般结构如下:

int main(int argc, char* argv[])
{

        //获得platform
        clGetPlatformIDs(1, &cpPlatform, NULL);

        //获得GPU设备
        clGetDeviceIDs(cpPlatform, CL_DEVICE_TYPE_GPU, 1, &cdDevice, NULL);

        //创建上下文环境
        hContext = clCreateContext(0, 1, &cdDevice, NULL, NULL, &ciErr);

        //创建命令队列
        cqCommandQueue = clCreateCommandQueue(hContext, cdDevice, 0, &ciErr);

        //在设备端开辟存储区
        cmDevSrc = clCreateBuffer(hContext, CL_MEM_READ_ONLY, sizeof(cl_float) * WorkSize, NULL, &ciErr);

        //创建program
        cpProgram = clCreateProgramWithSource(hContext, 1, &sProgramSource, NULL, NULL);

        //构建program
        clBuildProgram(cpProgram, 0, NULL, NULL, NULL, NULL);

        //创建运行在设备端的Kernel
        ckKernel = clCreateKernel(cpProgram, "VectorAdd", &ciErr);

        //为Kernel设置参数
        clSetKernelArg(ckKernel, 0, sizeof(cl_mem), (void*)&cmDevSrc);

        //将host主存数据copy到device端
        ciErr = clEnqueueWriteBuffer(cqCommandQueue, cmDevSrc, CL_FALSE, 0, sizeof(cl_float) * WorkSize, src, 0,

        // 调用Kernel,执行
        clEnqueueNDRangeKernel(cqCommandQueue, ckKernel, 1, NULL, (size_t*)&LocalSize, (size_t*)&WorkSize, 0, NULL, NULL);

       //从device端读取计算结果到host端
       clEnqueueReadBuffer(cqCommandQueue, cmDevDst, CL_TRUE, 0, sizeof(cl_float) * WorkSize, dst, 0, NULL, NULL);

        return 0;
}

 

在运行过程中,可能会因为某些参数的设置出了问题,导致GPU端不能正确运行。这时候最好调试一下,究竟在哪里出了问题,没有得到想要的结果,这样才能够准确定位。我就在的搭配cpProgram的时候,没有进行clBuildProgram,导致后面出现问题:(。

原创粉丝点击