7.6 统一地址空间

之前的OpenCL标准中,编程者常常需要为了不同地址空间的数据写很多版本的OpenCL C函数(其实就是参数中的地址描述符不同)。考虑下,下面两种数据数组,第一个在全局变量中,另一个缓存在局部内存中。两种方式都很简单,对于全局内存指针来说,函数可以直接使用(就像在有自动缓存系统的CPU上使用一样),另一种就是使用局部内存(GPU会将数组存储在快速便签式内存中)。OpenCL 2.0之前的标准中,要对这两种数组进行同样的操作,就需要将一个函数写两遍,如代码清单7.1所示。


1. void doDoubleGlobal(
2. __global float *data,
3. int index){

5. data[index] *= 2;
6. }

8. void doDoubleGlobal(
9. __local float *data,
10. int index){

12. data[index] *= 2;
13. }

15. __kernel
16. void doubleData(
17. global float *globalData, // the data
18. local float *localData, // local storage
19. int useLocal){ // whether or not to use local memory

21. int globalId = get_global_id(0);
22. int localId = get_local_id(0);

24. if (useLocal){
25. // copy data to local memroy
26. localData[localId] = globalData[globalId];

28. doDoubleLocal(localData, localId);

30. globalData[globalId] = localData[localId];
31. } else {
32. doDoubleGlobal(globalData, globalId);
33. }
34. }

代码清单7.1 在OpenCl 1.x中,需要对不同的寻址空间定义不同版本的函数

OpenCL 2.0开始,就不需要在这样定义函数了,同样的一个函数,可以通过统一内存地址覆盖所有内存空间。代码参见代码清单7.2。


1. void doDoubleGlobal(
2. float *data,
3. int index){

5. data[index] *= 2;
6. }

8. __kernel
9. void doubleData(
10. global float *globalData, // the data
11. local float *localData, // local storage
12. int useLocal){ // whether or not to use local memory

14. int globalId = get_global_id(0);
15. int localId = get_local_id(0);

17. generic float *data; // generic keyword not required
18. int myIndex;

20. if (useLocal){
21. // copy data to local memroy
22. localData[localId] = globalData[globalId];

24. // set data to local address space
25. data = localData;
26. myIndex = localId;
27. } else {
28. // set data to global address space
29. data = globalData;
30. myIndex = globalId;
31. }

33. doDouble(data, myIndex);

35. if (useLocal){
36. globalData[globalId] = localData[localId];
37. }
38. }

代码清单7.2 OpenCL 2.0中使用统一地址空间将7.1中代码重写

统一地址空间包括全局、局部和私有地址空间。2.0标准中,并未将常量地址划分到同一地址空间中。虽然,常量空间在逻辑上是全局便令的一部分,不过在某些处理器上(特别是图像处理器),常量数据会映射到特定的硬件单元上,并不能使用指令对其进行动态指定。