cuda_opencv 矩阵相加

 1 #include <stdlib.h>
 2 #include <stdio.h>
 3 #include <opencv/cv.h>
 4 #include <opencv/highgui.h>
 5 #include <opencv2/opencv.hpp>
 6 
 7 #include "cuda_runtime.h"
 8 #include "device_launch_parameters.h"
 9 using namespace std;
10 using namespace cv;
11 
12 __global__ void Add_kernel(const int2* d_A, const int2* d_B,int2*  d_C,int width, int height)
13 {
14     int x = threadIdx.x + blockIdx.x * blockDim.x;
15     int y = threadIdx.y + blockIdx.y * blockDim.y;
16 
17     if(x < width && y < height)
18     {
19         int offset = x + y*width;
20         d_C[offset].x = d_A[offset].x + d_B[offset].x;
21         d_C[offset].y = d_A[offset].y + d_B[offset].y;
22     }
23 }
24 int main()
25 {
26     Mat img(3, 4, CV_32S, Scalar_<int>(0));
27 
28     cout<<img<<endl;
29     cout<<endl;
30 
31 
32     for(int i = 0 ; i < img.rows; i++)
33     {
34         for(int j = 0 ; j < img.cols; j++)
35         {
36             img.at<int>(i,j)=i+j;
37         }
38     }
39     cout<<endl;
40 
41     cout<<img<<endl;
42 
43 
44     size_t memSize = img.step * img.rows;
45     int2* d_A = NULL;
46     int2* d_B = NULL;
47     int2* d_C = NULL;
48     cudaMalloc((void**)&d_A, memSize);
49     cudaMalloc((void**)&d_B, memSize);
50     cudaMalloc((void**)&d_C, memSize);
51 
52     cudaMemcpy(d_A,img.data,memSize, cudaMemcpyHostToDevice);
53     cudaMemcpy(d_B,img.data,memSize, cudaMemcpyHostToDevice);
54 
55     dim3 threads(16, 16);
56     dim3 grids((img.rows + threads.x - 1)/threads.x,(img.cols + threads.y - 1)/threads.y);
57     Add_kernel<<<grids,threads>>>(d_A, d_B, d_C, img.rows, img.cols);
58 
59     cudaMemcpy(img.data, d_C,memSize,cudaMemcpyDeviceToHost);
60     cout<<"GPU"<<endl;
61     cout<<img<<endl;
62     cudaFree(d_A);
63     cudaFree(d_B);
64     cudaFree(d_C);
65 
66     system("pause");
67     return 0;
68 }