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 }
