kaio-runtime 0.4.1

//! KAIO error types.

/// Errors that can occur during KAIO runtime operations.
#[derive(Debug, thiserror::Error)]
pub enum KaioError {
    /// A CUDA driver API call failed.
    #[error("CUDA driver error: {0}")]
    Cuda(#[from] cudarc::driver::DriverError),

    /// Device memory allocation failed.
    #[error("out of device memory: requested {requested} bytes")]
    OutOfMemory {
        /// Number of bytes requested.
        requested: usize,
    },

    /// Invalid kernel launch configuration.
    #[error("invalid kernel configuration: {0}")]
    InvalidConfig(String),

    /// No GPU device found at the given ordinal.
    #[error("device not found: ordinal {0}")]
    DeviceNotFound(usize),

    /// Failed to load a PTX module into the driver.
    #[error("PTX module load failed: {0}")]
    PtxLoad(String),

    /// A `PtxModule` failed validation before being handed to the driver.
    ///
    /// Raised by [`KaioDevice::load_module`](crate::KaioDevice::load_module)
    /// when the module's target SM is too low for a feature it uses
    /// (e.g. `mma.sync` in a `sm_70` module). Surfacing this before the
    /// driver compiles the PTX avoids cryptic downstream errors.
    #[error("PTX module validation failed: {0}")]
    Validation(#[from] kaio_core::ir::ValidationError),
}

/// Convenience alias for `std::result::Result<T, KaioError>`.
pub type Result<T> = std::result::Result<T, KaioError>;