libmir-metal 0.1.0

Metal inference backend for libmir
1
2
3
4
5
6
7
8
9
10
use super::{Array, Result, Stream};

pub fn export_once(logits: &Array, stream: &Stream) -> Result<()> {
    let Some(path) = stream.take_graph_dump_path() else {
        return Ok(());
    };
    logits.export_graph_dot(path)?;
    tracing::debug!(path = %path.display(), "exported MLX decode graph");
    Ok(())
}