cubecl-runtime 0.11.0

The runtime API of CubeCL: clients, devices, handles and everything a kernel launch needs.
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

//! The one payload every error type carries when the device is lost/poisoned.

use alloc::string::String;
use cubecl_environment::backtrace::BackTrace;
use thiserror::Error;

/// A fault the driver makes sticky: an illegal address, a trap, an ECC error, a
/// device lost outright. Nothing on the device can be trusted afterwards, and
/// every later call on it fails the same way until the process exits.
#[derive(Error, Clone)]
#[cfg_attr(serializable, derive(serde::Serialize, serde::Deserialize))]
#[error("{reason}\nBacktrace:\n{backtrace}")]
pub struct DevicePoison {
    /// The driver call that reported it.
    pub reason: String,
    /// The backtrace for this error.
    #[cfg_attr(serializable, serde(skip))]
    pub backtrace: BackTrace,
}

impl DevicePoison {
    /// The fault, with the backtrace captured here.
    pub fn new(reason: impl Into<String>) -> Self {
        Self {
            reason: reason.into(),
            backtrace: BackTrace::capture(),
        }
    }
}

impl core::fmt::Debug for DevicePoison {
    fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
        write!(f, "{self}")
    }
}