OpenCL 學習step by step (5) 使用二維NDRange workgroup


      在本教程中,我們使用二維NDRange來設置workgroup,這樣在opencl中,workitme的組織形式是二維的,Kernel中 的代碼也要做相應的改變,我們先看一下clEnqueueNDRangeKernel函數的變化。首先我們指定了workgroup size為localx*localy,通常這個值為64的倍數,但最好不要超過256。

//執行kernel,Range用2維,work itmes size為width*height,
cl_event ev;
size_t globalThreads[] = {width, height};
size_t localx, localy;
if(width/8 > 4)
    localx = 16;
else if(width < 8)
    localx = width;
else localx = 8;

if(height/8 > 4)
    localy = 16;
else if (height < 8)
    localy = height;
else localy = 8;

size_t localThreads[] = {localx, localy}; // localx*localy應該是64的倍數
printf("global_work_size =(%d,%d), local_work_size=(%d, %d)\n",width,height,localx,localy);

clTimer.Reset();
clTimer.Start();
clEnqueueNDRangeKernel( queue,
    kernel,
    2,
    NULL,
    globalThreads,
    localThreads, 0, NULL, &ev);

注意:在上面代碼中,定義global threads以及local threads數量,都是通過二維數組的方式進行的。

    新的Kernel代碼如下:

#pragma OPENCL EXTENSION cl_amd_printf : enable

__kernel void vecadd(__global const float* a, __global const float* b, __global float* c)
{
int x = get_global_id(0);
int y = get_global_id(1);
int width = get_global_size(0);
int height = get_global_size(1);
if(x == 1 && y ==1)
printf("%d, %d,%d,%d,%d,%d\n",get_local_size(0),get_local_size(1),get_local_id(0),get_local_id(1),get_group_id(0),get_group_id(1));

c[x + y * width] = a[x + y * width] + b[x + y * width];

}

      我們在kernel中增加了#pragma OPENCL EXTENSION cl_amd_printf : enable,以便在kernel中通過printf函數進行debug,這是AMD的一個擴展。printf還可以直接打印出float4這樣的向量,比如printf(“%v4f”, vec)。

      另外,在main.cpp中增加一行代碼:

//告訴driver dump il和isa文件
_putenv("GPU_DUMP_DEVICE_KERNEL=3");

      我們可以在程序目錄dump出il和isa形式的kernel文件,對於熟悉isa匯編的人,這是一個很好的調試performance的方法。

     在最新的app sdk 2.7及以后的sdk中,在kernel中使用printf的時候,這個程序會hang在哪兒,以前沒這種情況。

程序執行界面。

image

完整的代碼請參考:

工程文件gclTutorial4

代碼下載:

http://files.cnblogs.com/mikewolf2002/gclTutorial.zip


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM