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);

}