如何在OpenCL中获取当前调用主机线程的默认设备ID

OpenCL本身并没有提供和HIP中hipGetDevice()完全对应的内置函数，这是因为两者的模型设计不同：HIP（及CUDA）会为每个主机线程维护一个"默认设备"状态，而OpenCL的设备关联是通过**上下文（Context）和命令队列（Command Queue）**来实现的，线程本身没有全局绑定的默认设备。不过你可以通过以下两种方式实现类似需求：

1. 自行维护线程本地存储的设备状态 #

你可以利用C语言的线程本地存储（TLS）来手动跟踪当前线程使用的设备，模拟HIP的默认设备机制：

#include <CL/cl.h>

// 线程本地变量，存储当前线程关联的设备ID
static _Thread_local cl_device_id current_thread_device = NULL;

// 自定义函数：设置当前线程的默认设备
cl_int clSetCurrentDevice(cl_device_id device) {
    if (!device) {
        return CL_INVALID_DEVICE;
    }
    current_thread_device = device;
    return CL_SUCCESS;
}

// 自定义函数：获取当前线程的默认设备，对应hipGetDevice()
cl_int clGetCurrentDevice(cl_device_id *device) {
    if (!device) {
        return CL_INVALID_VALUE;
    }
    if (!current_thread_device) {
        return CL_DEVICE_NOT_FOUND;
    }
    *device = current_thread_device;
    return CL_SUCCESS;
}

使用时，每次切换设备调用clSetCurrentDevice()，需要获取时调用clGetCurrentDevice()即可。注意_Thread_local是C11标准关键字，部分编译器（如GCC）也支持__thread替代。

2. 通过现有OpenCL对象关联设备 #

如果你代码中已经在使用命令队列或上下文，可以直接从这些对象中提取对应的设备：

从命令队列获取关联设备 #

每个命令队列只能绑定到一个设备，通过clGetCommandQueueInfo可以获取该设备：

cl_int getDeviceFromQueue(cl_command_queue queue, cl_device_id *device) {
    if (!queue || !device) {
        return CL_INVALID_VALUE;
    }
    return clGetCommandQueueInfo(queue, CL_QUEUE_DEVICE, sizeof(cl_device_id), device, NULL);
}

从上下文获取关联设备 #

如果你的上下文是单设备上下文（多GPU场景下也常为每个设备创建单独上下文），可以从上下文提取设备：

cl_int getDeviceFromContext(cl_context context, cl_device_id *device) {
    if (!context || !device) {
        return CL_INVALID_VALUE;
    }
    cl_uint num_devices;
    cl_int err = clGetContextInfo(context, CL_CONTEXT_NUM_DEVICES, sizeof(cl_uint), &num_devices, NULL);
    if (err != CL_SUCCESS) {
        return err;
    }
    if (num_devices == 0) {
        return CL_DEVICE_NOT_FOUND;
    }
    // 提取上下文的第一个设备（单设备上下文即为目标设备）
    return clGetContextInfo(context, CL_CONTEXT_DEVICES, sizeof(cl_device_id), device, NULL);
}

内容的提问来源于stack exchange，提问作者Yogeesh Agarwal