cuda_oxide/
version.rs

1use std::fmt;
2
3use crate::{
4    error::{cuda_error, CudaResult},
5    sys, Cuda,
6};
7
8/// A CUDA device or API version
9#[derive(Clone, Copy, Debug)]
10pub struct CudaVersion {
11    pub major: u32,
12    pub minor: u32,
13}
14
15impl From<u32> for CudaVersion {
16    fn from(version: u32) -> Self {
17        CudaVersion {
18            major: version as u32 / 1000,
19            minor: (version as u32 % 1000) / 10,
20        }
21    }
22}
23
24impl Into<(u32, u32)> for CudaVersion {
25    fn into(self) -> (u32, u32) {
26        (self.major, self.minor)
27    }
28}
29
30impl From<(u32, u32)> for CudaVersion {
31    fn from(other: (u32, u32)) -> Self {
32        CudaVersion {
33            major: other.0,
34            minor: other.1,
35        }
36    }
37}
38
39impl fmt::Display for CudaVersion {
40    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
41        write!(f, "{}.{}", self.major, self.minor)
42    }
43}
44
45impl Cuda {
46    /// Gets the local driver version (not to be confused with device compute capability)
47    pub fn version() -> CudaResult<CudaVersion> {
48        let mut version = 0i32;
49        cuda_error(unsafe { sys::cuDriverGetVersion(&mut version as *mut i32) })?;
50        Ok((version as u32).into())
51    }
52}
cuda_oxide/version.rs

cuda_oxide/
version.rs
