Cuda C++ Thrust API与 Cuda Runtime API程序比较

今天买了本新书《高性能CUDA应用设计与开发方法与最佳实践》，今天读了第一章有点出获，分享给大家。

程序功能：给向量填充数据并计算各元素之和

1. CPU串行运行的代码：

//seqSerial.cpp：串行执行数组的填充及求和
#include<iostream>
#include<vector>
using namespace std;

int main()
{
 const int N=50000;
 //任务1：创建数组
 vector<int> a(N);
 //任务2：填充数组
 for(int i=0;i<N;i++)a[i]=i;
 //任务3：计算数组各元素之和
 int sumA=0;
 for(int i=0;i<N;i++)sumA+=a[i];
 //任务4：计算0-N-1之和
 int sumCheck=0;
 for(int i=0;i<N;i++)sumCheck+=i;
 //任务5：检查结果是否正确
 if(sumA==sumCheck) cout<<"Test Succeeded!"<<endl;
 else {cerr<<"TestFailed!"<<endl;return(1);}
 return (0);
}

 2.Cuda Thrust C++ API 程序

#include<iostream>
using namespace std;

#include<thrust\/reduce.h>
#include<thrust/sequence.h>
#include<thrust/host_vector.h>
#include<thrust/device_vector.h>

int main()
{
 const int N=50000;
 //任务1：创建数组
 thrust::device_vector<int>a(N);
 //任务2：填充数组，并行运算
 thrust::sequence(a.begin(),a.end(),0);
 //任务3：计算数组元素之和，并行计算
 int sumA=thrust::reduce(a.begin(),a.end(),0);
 //
 int sumCheck=0;
 for(int i=0;i<N;i++)
  sumCheck+=i;
 //
 if(sumA==sumCheck)cout<<"Test Succeeded!"<<endl;
 else
 {
  cerr<<"Test Failed!"<<endl;
  return(1);
 }
 getchar();
 return (0);
}

 

3.仅对数据填充改为Runtime API 程序

//使用cuda Runtime API完成向数组中填充连续整数
#include<iostream>
using namespace std;

#include<thrust\/reduce.h>
#include<thrust/sequence.h>
#include<thrust/host_vector.h>
#include<thrust/device_vector.h>

__global__ void fillKernel(int *a,int n)
{
 int tid=blockIdx.x*blockDim.x+threadIdx.x;
 if(tid<n) a[tid]=tid;
}

void fill(int *d_a,int n)
{
 int nThreadsPerBlock=512;
 //int nBlocks=n/nThreadsPerBlock+(n%nThreadsPerBlock)?1:0);
 int nBlocks=(n+nThreadsPerBlock)/nThreadsPerBlock;
 fillKernel<<<nBlocks,nThreadsPerBlock>>>(d_a,n);
}

int main()
{
 const int N=50000;
 //任务1：创建数组
 thrust::device_vector<int>a(N);
 //任务2：填充数组，使用Runtime API 填充数组
 fill(thrust::raw_pointer_cast(&a[0]),N);
 //任务3：计算数组元素之和，并行计算
 int sumA=thrust::reduce(a.begin(),a.end(),0);
 //任务4:计算0-N-1之和
 int sumCheck=0;
 for(int i=0;i<N;i++)
  sumCheck+=i;
 //任务5：检查结果的正确性
 if(sumA==sumCheck)cout<<"Test Succeeded!"<<endl;
 else
 {
  cerr<<"Test Failed!"<<endl;
  return(1);
 }
 getchar();
 return (0);
}