The Meson Build System
http://mesonbuild.com/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
21 lines
400 B
21 lines
400 B
#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(); |
|
}
|
|
|