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

//! Provides Cuda Driver API utility functionality.

use super::ffi::*;
use super::{Error, API};

impl API {
    /// Initialize the Cuda Driver API.
    ///
    /// must be called before any other function from the driver API.
    pub fn init() -> Result<(), Error> {
        Ok(unsafe { API::ffi_init() }?)
    }

    unsafe fn ffi_init() -> Result<(), Error> {
        const FLAGS: u32 = 0u32;
        match cuInit(FLAGS) {
            CUresult::CUDA_SUCCESS => Ok(()),
            CUresult::CUDA_ERROR_INVALID_VALUE => {
                Err(Error::InvalidValue("Invalid value provided."))
            }
            CUresult::CUDA_ERROR_INVALID_DEVICE => Err(Error::InvalidDevice("Invalid device.")),
            CUresult::CUDA_ERROR_NO_DEVICE => Err(Error::NoDevice(
                "Unable to find a CUDA device. Try run `nvidia-smi` on your console.",
            )),
            status => Err(Error::Unknown(
                "Unable to initialze the Cuda Driver API.",
                status as i32 as u64,
            )),
        }
    }
}