如何以多个GPU(OpenCL 1.1)以编程方式发现平台上的特定GPU?

时间:2011-11-16 02:48:26

标签: opencl

我的Mac Pro(OSX 10.7)有两个GPU。系统信息应用程序显示图形/显示的以下详细信息:

    ATI Radeon HD 5770:
       Bus: PCIe
       Slot:    Slot-1
       Vendor:  ATI (0x1002)
       Device ID:   0x68b8
       ...

    ATI Radeon HD 5770:
      Bus:  PCIe
      Slot: Slot-2
      Device ID:    0x68b8
      Displays:
        LED Cinema Display:
          Main Display: Yes
          ...

我想使用未连接到显示器的GPU在Java应用程序中进行计算,并对OpenCL 1.1进行低级绑定。如何以编程方式发现插槽1中的GPU设备?

从我的日志文件中显示设备信息查询的结果:

... Device ATI Radeon HD 5770[AMD]: vendorId[1021b00] ...
... Device ATI Radeon HD 5770[AMD]: vendorId[2021b00] ...

相关文章:How to match OpenCL devices with a specific GPU given PCI vendor, device and bus IDs in a multi-GPU system?

2 个答案:

答案 0 :(得分:4)

听起来您知道可以使用clGetDeviceIds获取系统中的设备,并且可以使用CL_DEVICE_NAME查询clGetDeviceInfo之类的内容。

不幸的是,我不认为OpenCL API目前有一种跨平台的方式来识别当前用于驱动显示器的计算设备。大多数情况下,人们希望获得此设备,以便他们可以使用相同的设备更快地进行OpenGL / OpenCL共享。在您的情况下,您想知道驱动显示器的设备是什么,以便忽略它

但是,有一种方法可以执行特定于Macintosh的方法。既然你提到你在Mac上,那么就是这个过程:

  1. 使用GPU设备创建OpenCL上下文。
  2. 向系统询问当前的OpenGL背景。
  3. 通过扩展程序(来自cl_gl_ext.h)询问OpenCL哪个设备正在驱动显示器。
  4. 使用供应商ID忽略该设备。
  5. 这是一个完整的程序,可以在Mac上执行此操作。我正在跑狮子。

    // compile with:
    // clang -o test test.c -framework GLUT -framework OpenGL -framework OpenCL
    #include <GLUT/glut.h>
    #include <OpenGL/OpenGL.h>
    #include <OpenGL/CGLDevice.h>
    #include <OpenCL/opencl.h>
    #include <OpenCL/cl_gl_ext.h>
    #include <stdio.h>
    
    int main (int argc, char const *argv[]) {
      int i;
      cl_int error;
    
      // We need to do *something* to create a GL context:
      glutInit( &argc, (char**)argv );
      glutInitDisplayMode( GLUT_RGB | GLUT_DOUBLE | GLUT_DEPTH );
      glutCreateWindow( "OpenCL <-> OpenGL Test" );
    
      // So we can ask CGL for it:
      CGLContextObj gl_context = CGLGetCurrentContext();
    
      CGLShareGroupObj share_group = CGLGetShareGroup(gl_context);
      cl_context_properties properties[] = { CL_CONTEXT_PROPERTY_USE_CGL_SHAREGROUP_APPLE, 
        (intptr_t)share_group, 0 };
      cl_context context = clCreateContext(properties, 0, NULL, 0, 0, &error);
    
      // And now we can ask OpenCL which particular device is being used by
      // OpenGL to do the rendering, currently:
      cl_device_id renderer;
      clGetGLContextInfoAPPLE(context, gl_context, 
        CL_CGL_DEVICE_FOR_CURRENT_VIRTUAL_SCREEN_APPLE, sizeof(renderer), 
        &renderer, NULL);
    
      cl_uint id_in_use;
      clGetDeviceInfo(renderer, CL_DEVICE_VENDOR_ID, sizeof(cl_uint), 
        &id_in_use, NULL);
    
      // Determine the number of devices:
      size_t size;
      cl_uint num_devices;
      clGetContextInfo(context, CL_CONTEXT_DEVICES, 0, NULL, &size);
    
      num_devices = size / sizeof(cl_device_id);
      cl_device_id devices[num_devices];
      clGetContextInfo(context, CL_CONTEXT_DEVICES, size, devices, NULL);
    
      // Now we have everything we need to use the device that IS NOT doing
      // rendering to the screen for our compute:
      char buf[128];
      cl_uint vendor_id;  
      for (i = 0; i < num_devices; i++) {
        clGetDeviceInfo(devices[i], CL_DEVICE_NAME, 128, buf, NULL);
        clGetDeviceInfo(devices[i], CL_DEVICE_VENDOR_ID, sizeof(cl_uint), &vendor_id, NULL);
        fprintf(stdout, "%s (%x)", buf, vendor_id);
        if (vendor_id == id_in_use) {
          fprintf(stdout, " [ in use by GL for display ]\n");
        } else {
          fprintf(stdout, " [ totally free for compute! ]\n");
        }      
      }
    
      clReleaseContext(context);
      return 0;
    }
    

    当我在iMac(一个GPU)上尝试这个时,我得到:

    ATI Radeon HD 6970M (1021b00) [ in use by GL for display ]
    

    但是当我通过ssh在远程盒子上尝试这个时:

    ATI Radeon HD 5770 (1021b00) [ totally free for compute! ]
    

    告诉我你的输出!我没有两个GPU盒子:)) 在我朋友的多GPU盒子上,运行Mac OS 10.7.2:

    GeForce GTX 285 (1022600) [ totally free for compute! ]
    GeForce GT 120 (2022600) [ in use by GL for display ] 
    

    请注意,可能有比GLUT更好的方法来启动和运行GL。但是GLUT并不是那么糟糕 - 你甚至不必在屏幕上显示一个窗口。这个程序没有。

答案 1 :(得分:1)

您可能对我的图书馆感兴趣:https://github.com/nbigaouette/oclutils/

我开发了该库来管理机器上的多个OpenCL设备。它将根据报告的“max_compute_unit”数自动对可用设备列表进行排序。这样,在我的机器上,它总是会拿起两个强大的Nvidia GTX 580中的一个,而不是驱动显示器的(蹩脚的)GT220。

它支持nvidia(仅限GPU),amd(CPU和/或GPU),intel(仅限CPU)和苹果(CPU和/或GPU)平台。

请注意,它无法区分哪两张相同的卡正在运行显示器,因此它不是您问题的完美解决方案。我可能会尝试整合詹姆斯的解决方案,因为这也是我感兴趣的东西。