hpt-common 0.1.3

An internal library for common utilities for hpt
Documentation 
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76

use std::panic::Location;

use thiserror::Error;

/// Device-related errors such as device not found, CUDA errors
#[derive(Debug, Error)]
pub enum DeviceError {
    /// Device not found
    #[error("Device {device} not found at {location}")]
    NotFound {
        /// Name or ID of the device that was not found
        device: String,
        /// Location where the error occurred
        location: &'static Location<'static>,
    },

    /// CUDA driver error
    #[error("CUDA driver error: {message} at {location}")]
    #[cfg(feature = "cuda")]
    CudaDriverError {
        /// Error message
        message: String,
        /// Source error
        #[source]
        source: Option<cudarc::driver::result::DriverError>,
        /// Location where the error occurred
        location: &'static Location<'static>,
    },

    #[cfg(feature = "cuda")]
    /// CUDA Cublas error
    #[error("CUDA Cublas error: {message} at {location}")]
    CudaCublasError {
        /// Error message
        message: String,
        /// Source error
        #[source]
        source: Option<cudarc::cublas::result::CublasError>,
        /// Location where the error occurred
        location: &'static Location<'static>,
    },

    /// Environment variable not set
    #[error("Environment variable {variable} not set at {location}")]
    EnvVarNotSet {
        /// Name of the environment variable that was not set
        variable: String,
        /// Location where the error occurred
        location: &'static Location<'static>,
    },
}

#[cfg(feature = "cuda")]
mod impls {
    use crate::error::base::TensorError;
    use crate::error::device::DeviceError;
    use std::panic::Location;
    impl From<cudarc::driver::result::DriverError> for TensorError {
        fn from(source: cudarc::driver::result::DriverError) -> Self {
            Self::Device(DeviceError::CudaDriverError {
                message: source.to_string(),
                source: Some(source),
                location: Location::caller(),
            })
        }
    }
    impl From<cudarc::cublas::result::CublasError> for TensorError {
        fn from(source: cudarc::cublas::result::CublasError) -> Self {
            Self::Device(DeviceError::CudaCublasError {
                message: source.to_string(),
                source: Some(source),
                location: Location::caller(),
            })
        }
    }
}