rk3588 ocl-icd-libopencl1 使用 OpenCL API 调用gpu

使用 OpenCL API 调用 GPU 进行计算，需要先编写 OpenCL 程序。OpenCL 程序是由 C 语言编写的，可以使用 OpenCL API 调用 OpenCL 驱动程序来执行计算任务。

以下是一个简单的 OpenCL 程序示例，可以将两个向量相加：

__kernel void vectorAdd(__global const float* a, __global const float* b, __global float* c)
{
    int i = get_global_id(0);
    c[i] = a[i] + b[i];
}

使用 OpenCL API 调用 OpenCL 驱动程序执行上述程序，以下是一个简单的代码示例：

#include <CL/opencl.h>

int main()
{
    cl_platform_id platform_id;
    cl_device_id device_id;
    cl_context context;
    cl_command_queue command_queue;
    cl_program program;
    cl_kernel kernel;
    cl_mem buffer_a, buffer_b, buffer_c;
    cl_int err;

    // 获取可用平台
    clGetPlatformIDs(1, &platform_id, NULL);

    // 获取可用设备
    clGetDeviceIDs(platform_id, CL_DEVICE_TYPE_GPU, 1, &device_id, NULL);

    // 创建 OpenCL 上下文
    context = clCreateContext(NULL, 1, &device_id, NULL, NULL, &err);

    // 创建命令队列
    command_queue = clCreateCommandQueue(context, device_id, 0, &err);

    // 创建 OpenCL 内核程序
    program = clCreateProgramWithSource(context, 1, &source_str, &source_size, &err);

    // 编译 OpenCL 程序
    clBuildProgram(program, 1, &device_id, NULL, NULL, NULL);

    // 创建内核对象
    kernel = clCreateKernel(program, "vectorAdd", &err);

    // 创建输入缓冲区
    buffer_a = clCreateBuffer(context, CL_MEM_READ_ONLY, sizeof(float) * N, NULL, &err);
    buffer_b = clCreateBuffer(context, CL_MEM_READ_ONLY, sizeof(float) * N, NULL, &err);

    // 创建输出缓冲区
    buffer_c = clCreateBuffer(context, CL_MEM_WRITE_ONLY, sizeof(float) * N, NULL, &err);

    // 将数据从主机内存拷贝到设备缓冲区
    clEnqueueWriteBuffer(command_queue, buffer_a, CL_TRUE, 0, sizeof(float) * N, a, 0, NULL, NULL);
    clEnqueueWriteBuffer(command_queue, buffer_b, CL_TRUE, 0, sizeof(float) * N, b, 0, NULL, NULL);

    // 设置内核参数
    clSetKernelArg(kernel, 0, sizeof(cl_mem), (void *)&buffer_a);
    clSetKernelArg(kernel, 1, sizeof(cl_mem), (void *)&buffer_b);
    clSetKernelArg(kernel, 2, sizeof(cl_mem), (void *)&buffer_c);

    // 执行内核
    size_t global_item_size = N;
    size_t local_item_size = 64; // 可以根据设备配置调整
    clEnqueueNDRangeKernel(command_queue, kernel, 1, NULL, &global_item_size, &local_item_size, 0, NULL, NULL);

    // 将数据从设备缓冲区拷贝到主机内存
    clEnqueueReadBuffer(command_queue, buffer_c, CL_TRUE, 0, sizeof(float) * N, c, 0, NULL, NULL);

    // 释放资源
    clReleaseMemObject(buffer_a);
    clReleaseMemObject(buffer_b);
    clReleaseMemObject(buffer_c);
    clReleaseKernel(kernel);
    clReleaseProgram(program);
    clReleaseCommandQueue(command_queue);
    clReleaseContext(context);

    return 0;
}

以上代码示例中，我们使用了 OpenCL API 创建了一个命令队列和一个内核对象，并将输入数据拷贝到设备缓冲区，最后执行内核程序并将输出数据拷贝到主机内存。