CUDA编程指南学习3.2.2
#CUDA C 版本
本文依照CUDA C编程指南9.0 3.2.2节内容进行代码完善
**
#代码实现&理解
**
//利用GPU实现乘法
#include
#include <stdlib.h>
#include <stdio.h>
//Device code
global void VecMuti(float *A,float *B,float *C,int N)
{
int i = blockDim.x * blockIdx.x +threadIdx.x;
if(i<N)
C[i] = A[i] * B[i];
}
//Host code
int main()
{
int N = 10;
size_t size = N*sizeof(float);
//Alloate input vectors h_A in host memory
float * h_A = (float*)malloc(size);
float * h_B = (float*)malloc(size);
float * h_C = (float*)malloc(size);
//Initialize input vectors
for(int i = 0;i<N;i++)
{
h_A[i] = i;
h_B[i] = -i;
}
//Allocate vectors in device memory
float * d_A;
float * d_B;
float * d_C;
cudaMalloc(&d_A,size);
cudaMalloc(&d_B,size);
cudaMalloc(&d_C,size);
//copy Host Memory to Device Memory
cudaMemcpy(d_A,h_A,size,cudaMemcpyHostToDevice);
cudaMemcpy(d_B,h_B,size,cudaMemcpyHostToDevice);
//Invoke kernel
int threadsPerBlock = 256;
int blocksPerGrid =
(N+threadsPerBlock-1)/threadsPerBlock;
VecMuti<<<blocksPerGrid,threadsPerBlock>>>(d_A,d_B,d_C,N);
//copy results from Device memory to Host memory
cudaMemcpy(h_C,d_C,size,cudaMemcpyDeviceToHost);
//Display results
for(int i = 0;i< N;i++)
{
printf("%d-----%f\n",i,h_C[i]);//在#include <stdio.h>文件中
}
//free Device memory
cudaFree(d_A);
cudaFree(d_B);
cudaFree(d_C);
//free Host memory;
free(h_A);
free(h_B);
free(h_C);
}