From af1a266670d040d2f4083ff309d732d648afba2a Mon Sep 17 00:00:00 2001 From: Angelos Mouzakitis Date: Tue, 10 Oct 2023 14:33:42 +0000 Subject: Add submodule dependency files Change-Id: Iaf8d18082d3991dec7c0ebbea540f092188eb4ec --- .../cuda/10 cuda dependency/modules/meson.build | 2 ++ .../cuda/10 cuda dependency/modules/prog.cc | 33 ++++++++++++++++++++++ 2 files changed, 35 insertions(+) create mode 100644 meson/test cases/cuda/10 cuda dependency/modules/meson.build create mode 100644 meson/test cases/cuda/10 cuda dependency/modules/prog.cc (limited to 'meson/test cases/cuda/10 cuda dependency/modules') 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 +#include +#include + +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; +} -- cgit 1.2.3-korg