aboutsummaryrefslogtreecommitdiffstats
path: root/meson/test cases/cuda/10 cuda dependency/modules
diff options
context:
space:
mode:
authorAngelos Mouzakitis <a.mouzakitis@virtualopensystems.com>2023-10-10 14:33:42 +0000
committerAngelos Mouzakitis <a.mouzakitis@virtualopensystems.com>2023-10-10 14:33:42 +0000
commitaf1a266670d040d2f4083ff309d732d648afba2a (patch)
tree2fc46203448ddcc6f81546d379abfaeb323575e9 /meson/test cases/cuda/10 cuda dependency/modules
parente02cda008591317b1625707ff8e115a4841aa889 (diff)
Add submodule dependency filesHEADmaster
Change-Id: Iaf8d18082d3991dec7c0ebbea540f092188eb4ec
Diffstat (limited to 'meson/test cases/cuda/10 cuda dependency/modules')
-rw-r--r--meson/test cases/cuda/10 cuda dependency/modules/meson.build2
-rw-r--r--meson/test cases/cuda/10 cuda dependency/modules/prog.cc33
2 files changed, 35 insertions, 0 deletions
diff --git a/meson/test cases/cuda/10 cuda dependency/modules/meson.build b/meson/test cases/cuda/10 cuda dependency/modules/meson.build
new file mode 100644
index 000000000..0da43f29d
--- /dev/null
+++ b/meson/test cases/cuda/10 cuda dependency/modules/meson.build
@@ -0,0 +1,2 @@
+exe = executable('prog', 'prog.cc', dependencies: dependency('cuda', modules: ['cublas']))
+test('cudatest', exe)
diff --git a/meson/test cases/cuda/10 cuda dependency/modules/prog.cc b/meson/test cases/cuda/10 cuda dependency/modules/prog.cc
new file mode 100644
index 000000000..b4af4d41f
--- /dev/null
+++ b/meson/test cases/cuda/10 cuda dependency/modules/prog.cc
@@ -0,0 +1,33 @@
+#include <cuda_runtime.h>
+#include <cublas_v2.h>
+#include <iostream>
+
+int cuda_devices(void) {
+ int result = 0;
+ cudaGetDeviceCount(&result);
+ return result;
+}
+
+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";
+
+ cublasHandle_t handle;
+ if (cublasCreate(&handle) != CUBLAS_STATUS_SUCCESS) {
+ std::cout << "cuBLAS initialization failed. Exiting.\n";
+ return -1;
+ }
+
+ std::cout << "Initialized cuBLAS\n";
+ if (cublasDestroy(handle) != CUBLAS_STATUS_SUCCESS) {
+ std::cout << "cuBLAS de-initialization failed. Exiting.\n";
+ return -1;
+ }
+
+ return 0;
+}