【发布时间】:2017-11-29 20:20:33
【问题描述】:
我是 cuda 的新手,我正在尝试用它来执行 Eratosthenes 的筛子。该代码适用于 1000000 以下的素数。在它上面我得到一个未知的内核启动错误。现在我明白这是因为我试图启动一个包含太多块的网格。但是,如果我将块设置为 1000,我不会得到所有的素数。我认为内核中的索引可能存在问题,但不确定。
#include "cuda_runtime.h"
#include "device_launch_parameters.h"
#include <iostream>
#include <fstream>
#include <stdio.h>
using namespace std;
__global__ static void Sieve(long * sieve, long sieve_size)
{
int idx = blockIdx.x * blockDim.x + threadIdx.x;
if (idx > 1) {
for (int i = idx+idx; i < sieve_size; i += idx) {
sieve[i] = 1;
}
}
}
int main()
{
long *device_sieve;
long *host_sieve = new long[4000000];
ofstream data("data.csv", ofstream::out);
double sieve_size = 4000000 / sizeof(long);
cudaSetDevice(0);
cudaDeviceSynchronize();
cudaThreadSynchronize();
cudaMalloc((void**)&device_sieve, sizeof(long) * sieve_size);
cudaError_t error1 = cudaGetLastError();
cout << "1" << cudaGetErrorString(error1) << endl;
int block = sqrt(sieve_size);
Sieve << <1, block >> >(device_sieve, sieve_size);
cudaThreadSynchronize();
cudaMemcpy(host_sieve, device_sieve, sizeof(long) * sieve_size, cudaMemcpyDeviceToHost);
cudaError_t error = cudaGetLastError();
cout << "2" << cudaGetErrorString(error) << endl;
cudaFree(device_sieve);
for (int i = 2; i < sieve_size; ++i)
if (host_sieve[i] == 0)
data << i << endl;
getchar();
cout << "DONE" << endl;
return 0;
}
【问题讨论】:
标签: c++ cuda sieve-of-eratosthenes