さまざまなブロックにわたる最終的な合計のアトミック計算を使用して、倍精度配列用の従来の内積カーネルを実装しようとしています。プログラミングガイドの116ページに記載されているように、倍精度でatomicAddを使用しました。おそらく、何か間違ったことをしています。すべてのブロックのスレッド全体の部分和は正しく計算されますが、その後、アトミック操作が正しく機能していないようです。同じデータでカーネルを実行するたびに、異なる結果を受け取ります。誰かが間違いを見つけたり、別の解決策を提供したりできれば幸いです。これが私のカーネルです:
__global__ void cuda_dot_kernel(int *n,double *a, double *b, double *dot_res)
{
__shared__ double cache[threadsPerBlock]; //thread shared memory
int global_tid=threadIdx.x + blockIdx.x * blockDim.x;
int i=0,cacheIndex=0;
double temp = 0;
cacheIndex = threadIdx.x;
while (global_tid < (*n)) {
temp += a[global_tid] * b[global_tid];
global_tid += blockDim.x * gridDim.x;
}
cache[cacheIndex] = temp;
__syncthreads();
for (i=blockDim.x/2; i>0; i>>=1) {
if (threadIdx.x < i) {
cache[threadIdx.x] += cache[threadIdx.x + i];
}
__syncthreads();
}
__syncthreads();
if (cacheIndex==0) {
*dot_res=cuda_atomicAdd(dot_res,cache[0]);
}
}
そして、これが私のデバイス関数atomicAddです:
__device__ double cuda_atomicAdd(double *address, double val)
{
double assumed,old=*address;
do {
assumed=old;
old= __longlong_as_double(atomicCAS((unsigned long long int*)address,
__double_as_longlong(assumed),
__double_as_longlong(val+assumed)));
}while (assumed!=old);
return old;
}