morok-schedule 0.1.0-alpha.2

Optimization passes and pattern engine for the Morok ML compiler
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

use snafu::Snafu;

#[derive(Debug, Clone, PartialEq, Eq, Snafu)]
#[snafu(visibility(pub))]
pub enum OptError {
    #[snafu(display("invalid argument type for operation; expected {expected}, found {found}"))]
    InvalidArgType { expected: &'static str, found: &'static str },
    #[snafu(display("operation validation failed for {op}: {reason}"))]
    ValidationFailed { op: &'static str, reason: &'static str },
    #[snafu(display("axis out of bounds: axis {axis} > max {max}"))]
    AxisOutOfBounds { axis: usize, max: usize },
    #[snafu(display("division constraint violated: {size} is not divisible by {amount}"))]
    DivisionError { size: usize, amount: usize },
    #[snafu(display("symbolic size cannot be verified for divisibility by {amount}"))]
    SymbolicDivisionError { amount: usize },
    #[snafu(display("expected Range operation, found other operation type"))]
    ExpectedRangeOperation,
    #[snafu(display("missing axis parameter for operation"))]
    MissingAxisParameter,
    #[snafu(display("backend doesn't support required feature: {feature}"))]
    UnsupportedFeature { feature: &'static str },
    #[snafu(display("optimization would exceed device limit: {limit_type} {value} > max {max}"))]
    DeviceLimitExceeded { limit_type: &'static str, value: usize, max: usize },
    // /// Tensor core pattern not matched.
    // TensorCoreNotMatched { reason: String },
}