您的位置:首页 > 其它

我写的cuda程序

2011-04-18 17:40 176 查看
 

#include "stdio.h"
#include "cuPrintf.cu"
#define N 10

 __global__ void add(int *da,int *db,int*dc)

{
  
          int tid=blockIdx.x;
          if(tid<N)
         c[tid]=a[tid]+b[tid];  
 
}
int main()
{     
 int a
,b
,c
;//定义host端的三个数组变量
 int *da,*db,*dc; //定义Debice端的三个数组指针变量

    /* 下面三个语句表示在设备端为三个计算数组分配存储空间*/
 cudaMalloc((void**)&da,N*sizeof(int));
 cudaMalloc((void**)&db,N*sizeof(int));
 cudaMalloc((void**))&dc,N*sizeof(int);
 /*在主机端输入依次输入两个数组变量*/
 printf("please input two arrays:");
 printf("the first array a
=");
          for(i=0;i<N)
  scanf("%d",&a[i]);
 printf("/n the second array b
=");
          for(i=0;i<N)
  scanf("%d",&b[i]);
 /*将Host段的数组内容传递到Debice端*/
 cudaMemcpy(da,a,N*sizeof(int),cudaHostToDevice);
 cudaMemcpy(db,b,N*sizeof(int),cudaHostToDevice);
 /*执行kernel函数,执行参数必须是设备端的参数,即da,db,dc 不应该是主机端的参数a,b,c*/
 add<<< N, 1 >>>(da,db,dc);
 /*将kernel函数执行后的结果由Device端传到Host端*/
 cudaMemcpy(c,dc,N*sizeof(int),cudaDeviceToHost);

 /*用for循环输出数组c[i]的结果*/
        for(i=0;i<N;i++)
       printf("the final array a[i]",c[i]);
 
 /*释放在Device端分配的显存空间*/
 cudaFree(da);
 cudaFree(db);
 
        return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息