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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
#[cfg(feature = "gpu")]
use crate::triton::cl;
use std::{error, fmt};
#[derive(Debug, Clone)]
#[cfg(any(feature = "gpu", feature = "opencl"))]
pub enum ClError {
DeviceNotFound,
PlatformNotFound,
BusIdNotAvailable,
NvidiaBusIdNotAvailable,
AmdTopologyNotAvailable,
PlatformNameNotAvailable,
CannotCreateContext,
CannotCreateQueue,
GetDeviceError,
}
#[cfg(any(feature = "gpu", feature = "opencl"))]
pub type ClResult<T> = std::result::Result<T, ClError>;
#[cfg(any(feature = "gpu", feature = "opencl"))]
impl fmt::Display for ClError {
fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
match self {
ClError::DeviceNotFound => write!(f, "Device not found."),
ClError::PlatformNotFound => write!(f, "Platform not found."),
ClError::BusIdNotAvailable => write!(f, "Cannot extract bus-id for the given device."),
ClError::NvidiaBusIdNotAvailable => {
write!(f, "Cannot extract bus-id for the given Nvidia device.")
}
ClError::AmdTopologyNotAvailable => {
write!(f, "Cannot extract bus-id for the given AMD device.")
}
ClError::PlatformNameNotAvailable => {
write!(f, "Cannot extract platform name for the given platform.")
}
ClError::CannotCreateContext => write!(f, "Cannot create cl_context."),
ClError::CannotCreateQueue => write!(f, "Cannot create cl_command_queue."),
ClError::GetDeviceError => write!(f, "Cannot get Device"),
}
}
}
#[derive(Debug, Clone)]
pub enum Error {
FullBuffer,
IndexOutOfBounds,
GPUError(String),
#[cfg(any(feature = "gpu", feature = "opencl"))]
ClError(ClError),
#[cfg(feature = "gpu")]
TritonError(String),
DecodingError,
Other(String),
}
#[cfg(feature = "gpu")]
impl From<ClError> for Error {
fn from(e: ClError) -> Self {
Self::ClError(e)
}
}
#[cfg(feature = "gpu")]
impl From<triton::Error> for Error {
fn from(e: triton::Error) -> Self {
Self::TritonError(e.to_string())
}
}
impl error::Error for Error {}
impl fmt::Display for Error {
fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
match self {
Error::FullBuffer => write!(
f,
"The size of the buffer cannot be greater than the hash arity."
),
Error::IndexOutOfBounds => write!(f, "The referenced index is outs of bounds."),
Error::GPUError(s) => write!(f, "GPU Error: {}", s),
#[cfg(any(feature = "gpu", feature = "opencl"))]
Error::ClError(e) => write!(f, "OpenCL Error: {}", e),
#[cfg(feature = "gpu")]
Error::TritonError(e) => write!(f, "Neptune-triton Error: {}", e),
Error::DecodingError => write!(f, "PrimeFieldDecodingError"),
Error::Other(s) => write!(f, "{}", s),
}
}
}