enki-anu 0.2.2

BorrowEngine, task recording, and pipeline synthesis for the Enki runtime
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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

pub mod cache;
pub mod compiler;
pub mod coordinator;
pub mod types;

use anyhow::Result;
pub use cache::{RAMTaskCache, calculate_compute_key, calculate_source_hash};
pub use coordinator::JitCoordinator;
pub use types::{ComputeSynthesisInput, TaskCompilationArtifact};

use crate::context::EnkiEngine;

pub struct PipelineSynthesizer {
    pub cache: RAMTaskCache,
}

impl PipelineSynthesizer {
    pub fn new() -> Self {
        Self {
            cache: RAMTaskCache::new(),
        }
    }

    pub fn synthesize_compute(
        &self,
        engine: &EnkiEngine,
        input: &ComputeSynthesisInput,
    ) -> Result<TaskCompilationArtifact> {
        let source_hash = calculate_source_hash(&input.nam_name);
        let cache_key = calculate_compute_key(
            source_hash,
            &input.nam_name,
            input.local_size,
            &input.arg_descriptors,
        );

        if let Some(cached_artifact) = self.cache.get(cache_key) {
            return Ok(cached_artifact);
        }

        JitCoordinator::orchestrate(engine, input)?;

        if let Some(cached_artifact) = self.cache.get(cache_key) {
            return Ok(cached_artifact);
        }

        Err(anyhow::anyhow!(
            "[PipelineSynthesizer] JIT Pipeline cache miss for nam '{}'. Please rebuild your project using 'cargo build'.",
            input.nam_name
        ))
    }
}

impl Default for PipelineSynthesizer {
    fn default() -> Self {
        Self::new()
    }
}