knok 0.1.0

Compile-time linalg graphs for Rust
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

#[derive(Clone, Copy, Debug)]
pub struct GraphArtifact {
    pub function_name: &'static str,
    pub input_shapes: &'static [&'static [usize]],
    pub output_shape: &'static [usize],
    pub variants: &'static [GraphArtifactVariant],
}

impl GraphArtifact {
    pub fn variant_for_driver(&self, driver: &str) -> Option<GraphArtifactVariant> {
        self.variants
            .iter()
            .copied()
            .find(|variant| variant.driver == driver)
    }

    pub fn first_variant(&self) -> Option<GraphArtifactVariant> {
        self.variants.first().copied()
    }
}

#[derive(Clone, Copy, Debug)]
pub struct GraphArtifactVariant {
    pub vmfb: &'static [u8],
    pub backend: &'static str,
    pub driver: &'static str,
    pub compile_flags: &'static [&'static str],
}