diff options
Diffstat (limited to 'meson/test cases/cuda/10 cuda dependency/version_reqs/prog.cc')
-rw-r--r-- | meson/test cases/cuda/10 cuda dependency/version_reqs/prog.cc | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/meson/test cases/cuda/10 cuda dependency/version_reqs/prog.cc b/meson/test cases/cuda/10 cuda dependency/version_reqs/prog.cc new file mode 100644 index 000000000..56688303f --- /dev/null +++ b/meson/test cases/cuda/10 cuda dependency/version_reqs/prog.cc @@ -0,0 +1,28 @@ +#include <cuda_runtime.h> +#include <iostream> + +int cuda_devices(void) { + int result = 0; + cudaGetDeviceCount(&result); + return result; +} + +int main(void) { + std::cout << "Compiled against CUDA version: " << CUDART_VERSION << "\n"; + int runtime_version = 0; + cudaError_t r = cudaRuntimeGetVersion(&runtime_version); + if (r != cudaSuccess) { + std::cout << "Couldn't obtain CUDA runtime version (error " << r << "). Exiting.\n"; + return -1; + } + std::cout << "CUDA runtime version: " << runtime_version << "\n"; + + 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; +} |