在测试套件中设置和拆除openCL单元测试的正确方法?

时间:2011-02-02 20:44:39

标签: java unit-testing installation opencl

快速说明:我正在使用JOCL和Java进行openCL开发。我认为我需要的openCL调用就像我只使用C或C ++一样。

我的问题是我希望能够运行我的每个测试,好像它是初始化后GPU运行的第一件事。这是我的代码:

protected cl_context clContext;
protected cl_command_queue commandQueue;

@Before
    public void setUp() {
            clContext = createContext();
            cl_device_id devices[] = getGPUDevices(clContext);
            commandQueue = clCreateCommandQueue(clContext, devices[0], 0, null);    
            CL.setExceptionsEnabled(true);
}

@After
public void tearDown() {
    clReleaseCommandQueue(commandQueue);
    clReleaseContext(clContext);
}

private cl_device_id[] getGPUDevices(cl_context clContext) {
    cl_device_id devices[]; 


    // Get the list of GPU devices associated with the context
    long numBytes[] = new long[1];
    clGetContextInfo(clContext, CL.CL_CONTEXT_DEVICES, 0, null, numBytes); 

    // Obtain the cl_device_id for the first device
    int numDevices = (int) numBytes[0] / Sizeof.cl_device_id;
    devices = new cl_device_id[numDevices];
    clGetContextInfo(clContext, CL_CONTEXT_DEVICES, numBytes[0],  
            Pointer.to(devices), null);

    return devices;
}

private cl_context createContext() {
    cl_context clContext;

    //System.out.println("Obtaining platform...");
    cl_platform_id platforms[] = new cl_platform_id[1];
    clGetPlatformIDs(platforms.length, platforms, null);
    cl_context_properties contextProperties = new cl_context_properties();
    contextProperties.addProperty(CL_CONTEXT_PLATFORM, platforms[0]);

    // Create an OpenCL context on a GPU device
    clContext = clCreateContextFromType(
            contextProperties, CL_DEVICE_TYPE_GPU, null, null, null);

    return clContext;
}

此代码在运行20多个测试后导致问题。出于某种原因,openCL将禁止CL_MEM_OBJECT_ALLOCATION_FAILURE。我修改了上面的代码,以便完全注释掉拆解,因此安装程序不会重新创建任何新的clContexts或commandQueues,现在我不会得到任何CL_MEM_OBJECT_ALLOCATION_FAILURE错误,无论我运行多少个测试。我不知道如何成功重置我的显卡状态,我错过了什么或做错了什么?请让我知道,谢谢。

1 个答案:

答案 0 :(得分:1)

也许是JOCL.org中的一个错误?我只是在http://jocl.jogamp.org的某些机器上运行了一些负载测试,但无法重现该问题。

@org.junit.Test
public void test(){
    for (int i = 0; i < 100000; i++) {
        CLContext context = CLContext.create();
        try{
            CLCommandQueue queue = context.getDevices()[0].createCommandQueue();
            queue.release();
        }finally{
            context.release();
        }
    }
}