【转】CUDA与二维动态数组

  1 /*
  2 * Copyright 徐洪志(西北农林科技大学.信息工程学院).  All rights reserved.
  3 * Data: 2012-4-22
  4 */
  5 //
  6 // 此程序是演示了二维动态数组空间申请和与显存数据相互拷贝的两种方法
  7 #include <stdio.h>
  8 //#include <cutil_inline.h>
  9 #include <iostream>
 10 #include <cuda_runtime.h>
 11 //#include <cutil.h>
 12 using namespace std;
 13 
 14 int main(int argc, char **argv)
 15 {
 16     
 17     //CUT_DEVICE_INIT(argc, argv);  // 启动 CUDA
 18 #if 1
 19     // 方法1.逐行拷贝
 20     float **CPU_ORIGN, **CPU_RET;  // host端原数据、拷贝回数据
 21     float **GPU;                   // device端数据
 22     int width = 5, height = 3;  // 数组的宽度和高度
 23     size_t size = sizeof(float)*width; // 数据的宽度in bytes
 24     int row, col;
 25 
 26     // 申请内存空间， 并初始化
 27     CPU_ORIGN = new float*[height];
 28     CPU_RET = new float*[height];
 29     for(row = 0; row < height; ++row)
 30     {
 31         CPU_ORIGN[row] = new float[width];
 32         CPU_RET[row] = new float[width];
 33         // 初始化数据
 34         for(col = 0; col < width; ++col)
 35         {
 36             CPU_ORIGN[row][col] = (float)(row + col);
 37             CPU_RET[row][col] = 0.0f;
 38         }
 39     }
 40 
 41     // 申请显存空间并初始化
 42     GPU = new float*[height];
 43     for(row = 0; row < height; ++row)
 44     {
 45           ( cudaMalloc((void**)&GPU[row], size));
 46           ( cudaMemset(GPU[row], 0, size));
 47     }
 48 
 49     // 将host端原数据拷贝到device端
 50     for(row = 0; row < height; ++row)
 51           (cudaMemcpy(GPU[row], CPU_ORIGN[row], size, cudaMemcpyHostToDevice));
 52 
 53     // 将device端数据拷贝到host端返回数据
 54     for(row = 0; row < height; ++row)
 55           (cudaMemcpy(CPU_RET[row], GPU[row], size, cudaMemcpyDeviceToHost));
 56 
 57     // 打印host端返回数据
 58     for(row = 0; row < height; ++row)
 59     {
 60         for(col = 0; col < width; ++col)
 61             cout << CPU_RET[row][col] << " ";
 62         cout << endl;
 63     }
 64     // 释放内存和显存空间
 65     free(CPU_ORIGN);
 66     free(CPU_RET);
 67     for(row = 0; row < height; ++row)
 68           (cudaFree(GPU[row]));
 69 #endif
 70 
 71 
 72 #if 0
 73     // 方法2.整体拷贝
 74     float **CPU_ORIGN, **CPU_RET;  // host端原数据、拷贝回数据
 75     float **GPU;                   // device端数据
 76     int width = 5, height = 3;         // 数组的宽度和高度
 77     size_t size = sizeof(float)*width; // 数据的宽度in bytes
 78     size_t pitch;
 79     int row, col;
 80 
 81     // 申请内存空间， 并初始化
 82     CPU_ORIGN = new float*[height];
 83     CPU_RET = new float*[height];
 84     for(row = 0; row < height; ++row)
 85     {
 86         CPU_ORIGN[row] = new float[width];
 87         CPU_RET[row] = new float[width];
 88         // 初始化数据
 89         for(col = 0; col < width; ++col)
 90         {
 91             CPU_ORIGN[row][col] = (float)(row + col);
 92             CPU_RET[row][col] = 0.0f;
 93         }
 94     }
 95 
 96     // 申请显存空间并初始化
 97       (cudaMallocPitch((void**)&GPU, &pitch, size, height));
 98       (cudaMemset2D(GPU, pitch, 0, size, height));
 99 
100     // 将host端原数据拷贝到device端
101       (cudaMemcpy2D(GPU, pitch, CPU_ORIGN, size, size, height, cudaMemcpyHostToDevice));
102 
103     // 将device端数据拷贝到host端返回数据
104       (cudaMemcpy2D(CPU_RET, size, GPU, pitch, size, height, cudaMemcpyDeviceToHost));
105 
106     // 打印host端返回数据
107     for(row = 0; row < height; ++row)
108     {
109         for(col = 0; col < width; ++col)
110             cout << CPU_RET[row][col] << " ";
111         cout << endl;
112     }
113     // 释放内存和显存空间
114     free(CPU_ORIGN);
115     free(CPU_RET);
116       (cudaFree(GPU));
117 #endif
118       system("pause");
119     //CUT_EXIT(argc, argv);  // 退出CUDA
120 }