blob: d2eff3f4b30f9640f4d4f4d81af307a1982eb783 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
#include <cuda.h>
#include <cuda_runtime.h>
#include <iostream>
#ifndef HOST_DEFINE
#error "HOST_DEFINE not defined!"
#endif
#ifndef PACKED_DEFINE
#error "PACKED_DEFINE not defined!"
#endif
static __global__ void DetermineIfValidCudaDevice()
{
}
#ifdef _MSC_VER
#pragma pack(push, 1)
#undef PACKED_DEFINE
#define PACKED_DEFINE
#endif
struct PACKED_DEFINE result_type
{
bool valid;
int value;
#if defined(NDEBUG) && !defined(DEFREL)
#error missing DEFREL flag
#endif
};
#ifdef _MSC_VER
#pragma pack(pop)
#endif
result_type can_launch_kernel()
{
result_type r;
DetermineIfValidCudaDevice<<<1, 1>>>();
r.valid = (cudaSuccess == cudaGetLastError());
if (r.valid) {
r.value = 1;
} else {
r.value = -1;
}
return r;
}
int main(int argc, char** argv)
{
cudaError_t err;
int nDevices = 0;
err = cudaGetDeviceCount(&nDevices);
if (err != cudaSuccess) {
std::cerr << cudaGetErrorString(err) << std::endl;
return 1;
}
return 0;
}
|
