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
357 B
21 lines
357 B
5 years ago
|
#include <cuda_runtime.h>
|
||
|
#include <iostream>
|
||
|
|
||
|
auto cuda_devices() {
|
||
|
int result = 0;
|
||
|
cudaGetDeviceCount(&result);
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
|
||
|
int main() {
|
||
|
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 0;
|
||
|
}
|