use singe_cuda_sys::driver;

use crate::{error::Result, try_cuda};

/// Returns the version number of the installed CUDA driver.
pub fn version() -> Result<i32> {
    let mut version: i32 = 0;
    unsafe {
        try_cuda!(driver::cuDriverGetVersion(&raw mut version))?;
    }
    Ok(version)
}

#[cfg(all(test, feature = "testing"))]
mod tests {
    use super::*;
    use crate::testing;

    #[test]
    fn it_works() {
        match version() {
            Ok(version) => assert_ne!(version, 0),
            Err(error) if testing::is_stub_library(&error) => {}
            Err(error) => panic!("{error:?}"),
        }
    }
}