爨爨爨好

  博客园  :: 首页  :: 新随笔  :: 联系 :: 订阅 订阅  :: 管理

▶ 表面内存使用

● 创建 cuda 数组时使用标志 cudaArraySurfaceLoadStore 来创建表面内存,可以用表面对象(surface object)或表面引用(surface reference)来对其进行读写。

● 使用 Surface Object API

■ 涉及的结构定义、接口函数。

1 // vector_types.h
2 struct __device_builtin__ __align__(4) uchar4
3 {
4     unsigned char x, y, z, w;
5 };
6 
7 // surface_types.h
8 typedef __device_builtin__ unsigned long long cudaSurfaceObject_t;

■ 完整的测试代码,使用表面内存进行简单的读写。

 1 #include <stdio.h>
 2 #include <stdlib.h>
 3 #include <malloc.h>
 4 #include <cuda_runtime_api.h>
 5 #include "device_launch_parameters.h"
 6 
 7 #define CEIL(x,y) (((x) + (y) - 1) / (y) + 1)
 8 
 9 __global__ void myKernel(cudaSurfaceObject_t inputSurfObj, cudaSurfaceObject_t outputSurfObj, int width, int height)
10 {
11     unsigned int idx = blockIdx.x * blockDim.x + threadIdx.x;
12     unsigned int idy = blockIdx.y * blockDim.y + threadIdx.y;
13     if (idx < width && idy < height)
14     {
15         uchar4 data;
16         // 简单的表面内存读写,使用了字节地址,而不是简单的线程编号
17         surf2Dread(&data, inputSurfObj, sizeof(float) * idx, idy);
18         surf2Dwrite(data, outputSurfObj, sizeof(float) * idx, idy);
19     }
20     cudaBindSurfaceToArray();
21 }
22 
23 int main()
24 {
25     // 基本数据
26     int i;
27     float *h_data, *d_data; 
28     int width = 32;
29     int height = 32;
30 
31     int size = sizeof(float)*width*height;
32     h_data = (float *)malloc(size);
33     cudaMalloc((void **)&d_data, size);
34 
35     for (i = 0; i < width*height; i++)
36         h_data[i] = (float)i;
37 
38     printf("\n\n");
39     for (i = 0; i < width*height; i++)
40     {
41         printf("%6.1f ", h_data[i]);
42         if ((i + 1) % width == 0)
43             printf("\n");
44     }
45     
46     // 申请 cuda 数组
47     cudaChannelFormatDesc channelDesc = cudaCreateChannelDesc(8, 8, 8, 8, cudaChannelFormatKindUnsigned);
48     cudaArray* cuInputArray;
49     cudaMallocArray(&cuInputArray, &channelDesc, width, height, cudaArraySurfaceLoadStore); 
50     cudaArray* cuOutputArray;
51     cudaMallocArray(&cuOutputArray, &channelDesc, width, height, cudaArraySurfaceLoadStore);
52     cudaMemcpyToArray(cuInputArray, 0, 0, h_data, size,cudaMemcpyHostToDevice);
53 
54     // 指定表面内存
55     struct cudaResourceDesc resDesc;
56     memset(&resDesc, 0, sizeof(resDesc));
57     resDesc.resType = cudaResourceTypeArray;
58 
59     // 创建表面对象
60     resDesc.res.array.array = cuInputArray;
61     cudaSurfaceObject_t inputSurfObj = 0;
62     cudaCreateSurfaceObject(&inputSurfObj, &resDesc);
63     resDesc.res.array.array = cuOutputArray;
64     cudaSurfaceObject_t outputSurfObj = 0;
65     cudaCreateSurfaceObject(&outputSurfObj, &resDesc);
66 
67     // 运行核函数
68     dim3 dimBlock(16, 16);
69     dim3 dimGrid(CEIL(width, dimBlock.x), CEIL(height, dimBlock.y));
70     myKernel << <dimGrid, dimBlock >> > (inputSurfObj, outputSurfObj, width, height);
71 
72     // 结果回收和检查结果
73     memset(h_data,0,size);// 刷掉原来的 h_data,再用 cuOutputArray 的数据写入     
74     cudaMemcpyFromArray(h_data, cuOutputArray, 0, 0, size, cudaMemcpyDeviceToHost);
75 
76     printf("\n\n");
77     for (i = 0; i < width*height; i++)
78     {
79         printf("%6.1f ", h_data[i]);
80         if ((i + 1) % width == 0)
81             printf("\n");
82     }
83 
84     // 回收工作
85     cudaDestroySurfaceObject(inputSurfObj);
86     cudaDestroySurfaceObject(outputSurfObj);
87     cudaFreeArray(cuInputArray);
88     cudaFreeArray(cuOutputArray);
89 
90     getchar();
91     return 0;
92 }

 

● 使用 Surface Reference API。

■ 表面引用的一些只读属性需要在声明的时候指定,以便编译时提前确定,只能在全局作用域内静态指定,不能作为参数传递给函数。使用 surface 指定纹理引用属性,Datatype 为数据类型,Type 为纹理引用类型,有 7 种,默认 cudaSurfaceType1D。

 1 surface<void, Type> surfRef;
 2 
 3 // cuda_texture_types.h
 4 template<class T, int dim = 1>
 5 struct __device_builtin_surface_type__ surface : public surfaceReference
 6 {
 7 #if !defined(__CUDACC_RTC__)
 8     __host__ surface(void)
 9     {
10         channelDesc = cudaCreateChannelDesc<T>();
11     }
12 
13     __host__ surface(struct cudaChannelFormatDesc desc)
14     {
15         channelDesc = desc;
16     }
17 #endif /* !__CUDACC_RTC__ */  
18 };
19 
20 //surface_types.h
21 #define cudaSurfaceType1D              0x01
22 #define cudaSurfaceType2D              0x02
23 #define cudaSurfaceType3D              0x03
24 #define cudaSurfaceTypeCubemap         0x0C
25 #define cudaSurfaceType1DLayered       0xF1
26 #define cudaSurfaceType2DLayered       0xF2
27 #define cudaSurfaceTypeCubemapLayered  0xFC
28 
29 // 访问边界模式
30 enum __device_builtin__ cudaSurfaceBoundaryMode
31 {
32     cudaBoundaryModeZero = 0,    // 0 边界模式
33     cudaBoundaryModeClamp = 1,   // 挤压模式
34     cudaBoundaryModeTrap = 2     // 陷阱模式
35 };
36 
37 // ?表面格式模式
38 enum __device_builtin__  cudaSurfaceFormatMode
39 {
40     cudaFormatModeForced = 0,   // 强制模式
41     cudaFormatModeAuto = 1      // 自动模式
42 };
43 
44 // 表面引用的通道描述
45 struct __device_builtin__ surfaceReference
46 {
47     struct cudaChannelFormatDesc channelDesc;
48 };
49 
50 // cuda_runtime_api.h
51 extern __host__ cudaError_t CUDARTAPI cudaBindSurfaceToArray(const struct surfaceReference *surfref, cudaArray_const_t array, const struct cudaChannelFormatDesc *desc);

 

■ 表面引用使用字节地址来定位访问(而不是像纹理那样使用 fetch 函数),如以上代码中 surf1Dread(surfRef, sizeof(float) * idx) 或是 surf1Dread(surfRef, sizeof(float) * idx) 。

■ 表面引用必须用函数 cudaBindSurfaceToArray() 绑定到 cuda 数组上才能使用,要求表面引用的维度、数据类型与该数组匹配,否则操作时未定义的,使用完后不需要特殊函数来解除绑定。

■ 将表面引用绑定到 cuda 数组上的范例代码。

 1 // 准备工作
 2 surface<void, Type>surfRef;
 3 
 4 ...
 5 
 6 int width, height;
 7 size_t pitch;
 8 float *d_data;
 9 cudaMallocPitch((void **)&d_data, &pitch, sizeof(float)*width, height);
10 
11 // 第一种方法,低层 API
12 surfaceReference* surfRefPtr;
13 cudaGetSurfaceReference(&surfRefPtr, "surfRef");
14 cudaChannelFormatDesc channelDesc;
15 cudaGetChannelDesc(&channelDesc, cuArray);
16 cudaBindSurfaceToArray(surfRef, cuArray, &channelDesc);
17 
18 // 第二种方法,高层 API
19 cudaBindSurfaceToArray(surfRef, cuArray);

 

■ 完整的应用样例代码。与前面表面对象代码的功能相同。

 1 #include <stdio.h>
 2 #include <stdlib.h>
 3 #include <malloc.h>
 4 #include <cuda_runtime_api.h>
 5 #include "device_launch_parameters.h"
 6 
 7 #define CEIL(x,y) (((x) + (y) - 1) / (y) + 1)
 8 
 9 // 声明表面引用
10 surface<void, 2> inputSurfRef;
11 surface<void, 2> outputSurfRef;
12 
13 __global__ void myKernel(int width, int height)
14 {
15     unsigned int idx = blockIdx.x * blockDim.x + threadIdx.x;
16     unsigned int idy = blockIdx.y * blockDim.y + threadIdx.y;
17     if (idx < width && idy < height)
18     {
19         uchar4 data;
20         // 简单的表面内存读写,使用了字节地址,而不是简单的线程编号
21         surf2Dread(&data, inputSurfRef, sizeof(float) * idx, idy);
22         surf2Dwrite(data, outputSurfRef, sizeof(float) * idx, idy);
23     }
24 }
25 
26 int main()
27 {
28     // 基本数据
29     int i;
30     float *h_data, *d_data; 
31     int width = 32;
32     int height = 32;
33 
34     int size = sizeof(float)*width*height;
35     h_data = (float *)malloc(size);
36     cudaMalloc((void **)&d_data, size);
37 
38     for (i = 0; i < width*height; i++)
39         h_data[i] = (float)i;
40 
41     printf("\n\n");
42     for (i = 0; i < width*height; i++)
43     {
44         printf("%6.1f ", h_data[i]);
45         if ((i + 1) % width == 0)
46             printf("\n");
47     }
48     
49     // 申请 cuda 数组
50     cudaChannelFormatDesc channelDesc = cudaCreateChannelDesc(8, 8, 8, 8, cudaChannelFormatKindUnsigned);
51     cudaArray* cuInputArray;
52     cudaMallocArray(&cuInputArray, &channelDesc, width, height, cudaArraySurfaceLoadStore); 
53     cudaArray* cuOutputArray;
54     cudaMallocArray(&cuOutputArray, &channelDesc, width, height, cudaArraySurfaceLoadStore);
55     cudaMemcpyToArray(cuInputArray, 0, 0, h_data, size,cudaMemcpyHostToDevice);
56 
57     // 绑定表面引用,注意与表面对象的使用不一样
58     cudaBindSurfaceToArray(inputSurfRef, cuInputArray);
59     cudaBindSurfaceToArray(outputSurfRef, cuOutputArray);
60 
61     // 运行核函数
62     dim3 dimBlock(16, 16);
63     dim3 dimGrid(CEIL(width, dimBlock.x), CEIL(height, dimBlock.y));
64     myKernel << <dimGrid, dimBlock >> > (width, height);
65 
66     // 结果回收和检查结果
67     memset(h_data,0,size);// 刷掉原来的 h_data,再用 cuOutputArray 的数据写入     
68     cudaMemcpyFromArray(h_data, cuOutputArray, 0, 0, size, cudaMemcpyDeviceToHost);
69 
70     printf("\n\n");
71     for (i = 0; i < width*height; i++)
72     {
73         printf("%6.1f ", h_data[i]);
74         if ((i + 1) % width == 0)
75             printf("\n");
76     }
77 
78     // 回收工作
79     cudaFreeArray(cuInputArray);
80     cudaFreeArray(cuOutputArray);
81 
82     getchar();
83     return 0;
84 }

 

▶ 立方体表面 Cubemap Surface 。 (想象成一个正方体的外表面)

● 一种特殊的二维分层表面。函数 surfCubemapread() 和函数 surfCubemapwrite() 来对其进行读写,使用一个整数下标和两个浮点数有序组来定义层号和表面坐标。

 

▶ 分层立方体表面 Cubemap Layered Surfaces 。(想象成一个多层的正方体的各外表面)

● 一种特殊的二维分层表面。函数 surfCubemapread() 和函数 surfCubemapwrite() 来对齐进行读写。使用一个整数下标和两个浮点数有序组来定义层号和表面坐标。

● 分层立方体贴图纹理只能使用函数 cudaMAlloc3DArray() 加上 cudaArrayLayered 和 cudaArrayCubemap 标志来声明,使用函数 texCubemapLayered() 来进行访问滤波只在同一层内部进行,不会跨层执行。

 

▶ cuda 数组。

● cuda 优化的数组类型,可以有一维或二维或三维,每个元素可以有 1 个或 2 个或 4 个分量,各分量可以是 1 B 或 2 B 或 4 B 尺寸的有符号或无符号整数,或 2 B 或 4 B 尺寸的浮点数。cuda 数组只能用纹理访问函数来访问,或表面函数来进行读写。

● 纹理内存和表面内存都是可缓存的,且不能保证缓存和内存的一致性。同一个核函数中,用纹理访问或表面访问来读取“已经全局写入或表面写入的内存”是未定义的。

 

▶ 压缩版的 surface_types.h

 1 #if !defined(__SURFACE_TYPES_H__)
 2 #define __SURFACE_TYPES_H__
 3 
 4 #include "driver_types.h"
 5 
 6 #define cudaSurfaceType1D              0x01
 7 #define cudaSurfaceType2D              0x02
 8 #define cudaSurfaceType3D              0x03
 9 #define cudaSurfaceTypeCubemap         0x0C
10 #define cudaSurfaceType1DLayered       0xF1
11 #define cudaSurfaceType2DLayered       0xF2
12 #define cudaSurfaceTypeCubemapLayered  0xFC
13 
14 //CUDA Surface boundary modes
15 enum __device_builtin__ cudaSurfaceBoundaryMode
16 {
17     cudaBoundaryModeZero = 0,   // Zero boundary mode */
18     cudaBoundaryModeClamp = 1,  // Clamp boundary mode */
19     cudaBoundaryModeTrap = 2    // Trap boundary mode */
20 };
21 
22 //CUDA Surface format modes
23 enum __device_builtin__  cudaSurfaceFormatMode
24 {
25     cudaFormatModeForced = 0,   // Forced format mode */
26     cudaFormatModeAuto = 1      // Auto format mode */
27 };
28 
29 //CUDA Surface reference
30 struct __device_builtin__ surfaceReference
31 {
32     // Channel descriptor for surface reference
33     struct cudaChannelFormatDesc channelDesc;
34 };
35 
36 //An opaque value that represents a CUDA Surface object
37 typedef __device_builtin__ unsigned long long cudaSurfaceObject_t;
38 
39 
40 #endif

 

posted on 2017-11-10 23:20  爨爨爨好  阅读(1630)  评论(0编辑  收藏  举报