meson/test cases/cuda/10 cuda dependency/modules/prog.cc


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
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;
}