Cudaの2つの数の和を求める演算
922 ワード
1.cudaプログラムの概要
Cudaはnvidia社が発表した計算性能を向上させるためにGPU上で高速演算を実行するツールで、GPUを用いて演算するため、プログラム実行時にCPUとGPU上のデータ共有を両立させなければならない.このセクションでは、最も簡単な2つの数を加算するプログラムについてのみ説明します.
2.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;
}