Cudaの2つの数の和を求める演算

922 ワード

1.cudaプログラムの概要
Cudaはnvidia社が発表した計算性能を向上させるためにGPU上で高速演算を実行するツールで、GPUを用いて演算するため、プログラム実行時にCPUとGPU上のデータ共有を両立させなければならない.このセクションでは、最も簡単な2つの数を加算するプログラムについてのみ説明します.
2.cudaコード
#include<iostream>
#include<cstdlib>
using namespace std;
#define N 1
__global__ void add(int *a,int *b,int *c) {
	*c = *a + *b;
}

int main() {
	int a,b,c;
 	a=1;
	b=9;
	int *d_a,*d_b,*d_c;
        /* allocate memory on device */
	cudaMalloc((void **)&d_a,N*sizeof(int));	
	cudaMalloc((void **)&d_b,N*sizeof(int));	
	cudaMalloc((void **)&d_c,N*sizeof(int));	
	cudaMemcpy(d_a,&a,N*sizeof(int),cudaMemcpyHostToDevice);
	cudaMemcpy(d_b,&b,N*sizeof(int),cudaMemcpyHostToDevice);
	add<<<1,1>>>(d_a,d_b,d_c);
	cudaMemcpy(&c,d_c,N*sizeof(int),cudaMemcpyDeviceToHost);
	cout<<"c:"<<c<<endl;
	cudaFree(d_a);
	cudaFree(d_b);
	cudaFree(d_c);
	return 0;
}