我正在尝试运行一段代码,使用Cuda中的并行缩减来查找数组的最大元素
#include "cuda_runtime.h"
#include "device_launch_parameters.h"
#include <stdio.h>
/* a is the array that holds the values and c is the array used to store the maximum in a block */
cudaError_t reduce_max(int *a,int *c,int size);
/*The kernel that performs the reduction */
__global__ void global_max(int *d_c, int * d_a)
{
int myId=threadIdx.x+blockDim.x*blockIdx.x;
int tid=threadIdx.x;
for(int s=(blockDim.x)/2; s>0; s>>1)
{
if(tid<s)
{
d_a[myId]=max(d_a[myId],d_a[myId+s]);
}
__syncthreads();
}
if(tid==0)
{
d_c[blockIdx.x]=d_a[myId];
}
}
int main()
{
const int arraySize = 1024;
int i;
int a[arraySize];
for(i=0;i<arraySize;i++)
{
a[i]=i;
}
int c[arraySize];
cudaError_t cudaStatus = reduce_max(a,c,arraySize);
if (cudaStatus != cudaSuccess) {
fprintf(stderr, "The required operation failed");
return 1;
}
cudaStatus = cudaThreadExit();
if (cudaStatus != cudaSuccess) {
fprintf(stderr, "cudaThreadExit failed!");
return 1;
}
return 0;
}
// Helper function for using CUDA to add vectors in parallel.
cudaError_t reduce_max(int *a,int *c,int size)
{
int *dev_a = 0;
int *dev_c = 0;
/*
dev_a and dev_c are the arrays on the device
*/
cudaError_t cudaStatus;
const dim3 blockSize(64,1,1);
const dim3 gridSize(size/blockSize.x,1,1);
// Choose which GPU to run on, change this on a multi-GPU system.
cudaStatus = cudaSetDevice(0);
if (cudaStatus != cudaSuccess) {
fprintf(stderr, "cudaSetDevice failed! Do you have a CUDA-capable GPU installed?");
goto Error;
}
/*Allocating the memory on the device */
cudaStatus = cudaMalloc((void**)&dev_c, size * sizeof(int));
if (cudaStatus != cudaSuccess) {
fprintf(stderr, "cudaMalloc failed!");
goto Error;
}
cudaStatus = cudaMalloc((void**)&dev_a, size * sizeof(int));
if (cudaStatus != cudaSuccess) {
fprintf(stderr, "cudaMalloc failed!");
goto Error;
}
/*Copying array from host to device */
cudaStatus = cudaMemcpy(dev_a, a, size * sizeof(int), cudaMemcpyHostToDevice);
if (cudaStatus != cudaSuccess) {
fprintf(stderr, "cudaMemcpy failed!");
goto Error;
}
/*Calling the kernel */
global_max<<<gridSize,blockSize>>>(dev_c, dev_a);
cudaStatus = cudaThreadSynchronize();
if (cudaStatus != cudaSuccess) {
fprintf(stderr, "cudaThreadSynchronize returned error code %d\n", cudaStatus);
goto Error;
}
// Copy output vector from GPU buffer to host memory.
cudaStatus = cudaMemcpy(c, dev_c, size * sizeof(int), cudaMemcpyDeviceToHost);
if (cudaStatus != cudaSuccess) {
fprintf(stderr, "cudaMemcpy failed!");
goto Error;
}
Error:
cudaFree(dev_c);
cudaFree(dev_a);
return cudaStatus;
}然而,在执行上面的代码时,我得到了错误: cudaThreadSynchronize返回了错误代码6。
我不能弄清楚这个问题。
发布于 2017-01-24 02:49:13
你的代码将永远运行。因此,您将遇到超时。
这一行被破坏了,你的编译器应该抛出一个警告:
for(int s=(blockDim.x)/2; s>0; s>>1)s>>1不会修改s变量。我很确定你指的是修改了s的s>>=1。如果不修改s,您的循环将永远运行,因此您会遇到内核超时。
改为执行以下操作:
for(int s=(blockDim.x)/2; s>0; s>>=1)https://stackoverflow.com/questions/41813252
复制相似问题