blob: 7b06570f8e679914c25eed64af210deb28363f96 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
#include <cuda_runtime.h>
#include <iostream>
auto cuda_devices(void) {
int result = 0;
cudaGetDeviceCount(&result);
return result;
}
int do_cuda_stuff();
int main(void) {
int n = cuda_devices();
if (n == 0) {
std::cout << "No Cuda hardware found. Exiting.\n";
return 0;
}
std::cout << "Found " << n << "Cuda devices.\n";
return do_cuda_stuff();
}
|