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 }
