aboutsummaryrefslogtreecommitdiffstats
path: root/meson/test cases/cuda/10 cuda dependency/version_reqs/prog.cc
diff options
context:
space:
mode:
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.cc28
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;
+}