vyre-self-substrate 0.6.1

Vyre self-substrate: vyre using its own primitives on its own scheduler problems. The recursion-thesis layer between vyre-primitives and vyre-driver.
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

//! Path-reconstruction substrate consumer.
//!
//! Wires `vyre_primitives::graph::path_reconstruct` so the optimizer can
//! recover an explicit walk from a parent vector. Used by call-graph
//! diagnostics (which path led from entry to a region flagged by an analysis
//! pass), megakernel chain reconstruction, and schedule-explanation telemetry.
//!
//! Per the primitive's spec: walks parent links from `target` back to
//! the root (a node whose parent points at itself), writing the
//! materialized path into a caller-provided scratch buffer and
//! returning its length.

mod dispatch;

pub use dispatch::{
    path_to_root_via, reconstruct_path_via, reconstruct_path_via_with_scratch,
    reconstruct_paths_via, reconstruct_paths_via_with_scratch_into,
};

use crate::graph::dispatch_bridge::{CachedProgram, ProgramCache};
#[cfg(test)]
use vyre_foundation::ir::Program;
use vyre_primitives::graph::path_reconstruct::PathReconstructStaticInputKey;

#[cfg(test)]
mod reference;
#[cfg(test)]
use reference::path_reconstruct_cpu;
#[cfg(test)]
pub use reference::{path_to_root, reference_reconstruct_path};

/// Caller-owned GPU dispatch scratch for path reconstruction.
#[derive(Debug, Default)]
pub struct PathReconstructGpuScratch {
    inputs: Vec<Vec<u8>>,
    len_out: Vec<u32>,
    static_input_key: Option<PathReconstructStaticInputKey>,
    single_program_cache: ProgramCache<u32, CachedSinglePathProgram>,
    batched_program_cache: ProgramCache<(u32, u32), CachedBatchedPathProgram>,
}

type CachedSinglePathProgram = CachedProgram;
type CachedBatchedPathProgram = CachedProgram;

impl PathReconstructGpuScratch {
    #[cfg(test)]
    fn single_program_builds(&self) -> usize {
        self.single_program_cache.builds()
    }

    #[cfg(test)]
    fn batched_program_builds(&self) -> usize {
        self.batched_program_cache.builds()
    }
}

#[cfg(test)]
mod tests;