cuda_opencv之向量相加

实现向量相加。。。实现与 矩阵相加一样~

 1 #include <stdlib.h>
 2 #include <stdio.h>
 3 #include <opencv/cv.hpp>
 4 #include <opencv/highgui.h>
 5 
 6 #include <cuda_runtime.h>
 7 #include <device_launch_parameters.h>
 8 
 9 using namespace std;
10 using namespace cv;
11 
12 #define N 10
13 
14 __global__ void Add_kernel(const int2* d_a, const int2* d_b,int2*  d_c ,int pp)
15 {
16     int x=threadIdx.x;
17     
18     if(x < pp)
19     {
20         d_c[x].x=d_a[x].x + d_b[x].x;
21         d_c[x].y=d_a[x].y + d_b[x].y;
22 
23     }
24 
25 }
26 int main()
27 {
28     Mat a(1,N,CV_32S,Scalar_<int>(0));
29     Mat b(1,N,CV_32S,Scalar_<int>(10));
30 
31     cout<<a<<endl;
32     cout<<endl;
33     cout<<b<<endl;
34     cout<<endl;
35 
36     size_t memSize = a.step * a.rows;
37 
38     int2* d_a = NULL;
39     int2* d_b = NULL;
40     int2* d_c = NULL;
41 
42     cudaMalloc((void**)&d_a,memSize);
43     cudaMalloc((void**)&d_b,memSize);
44     cudaMalloc((void**)&d_c,memSize);
45 
46     cudaMemcpy(d_a,a.data,memSize,cudaMemcpyHostToDevice);
47     cudaMemcpy(d_b,b.data,memSize,cudaMemcpyHostToDevice);
48 
49     Add_kernel<<<1,N>>>(d_a, d_b, d_c, N);
50     cudaMemcpy(a.data, d_c,memSize, cudaMemcpyDeviceToHost);
51     cout<<endl;
52     cout<<a<<endl;
53     system("pause");
54     return 0;
55 }

 

转载于:https://www.cnblogs.com/LzKlyhPorter/p/4611395.html