libmir-metal 0.3.0

Metal inference backend for libmir
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

use models::semantic::{DecoderLayerSpec, SemanticModelSpec};

use super::{Error, Result};

#[cfg(test)]
mod tests;

pub use architecture::metal::{
    FeedForwardLowering, LayerLowering, MetalDecoderPlan as DecoderLowering,
    MetalDecoderRuntime as DecoderRuntime, MixerLowering, NormalizationLowering,
};

pub fn plan(spec: &SemanticModelSpec) -> Result<DecoderLowering> {
    match architecture::metal::plan(spec) {
        Ok(plan) => Ok(plan),
        Err(error) => Err(Error::InvalidModel(error.to_string())),
    }
}

pub fn lower_layer(layer: &DecoderLayerSpec) -> Result<LayerLowering> {
    match architecture::metal::lower_layer(layer) {
        Ok(layer) => Ok(layer),
        Err(error) => Err(Error::InvalidModel(error.to_string())),
    }
}