sim-lib-interference-compute 0.1.0

Normalized tile-local f32 Tensor lowering for coherent interference.
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95

//! Resident result assembly over the canonical Tensor lowering.

use sim_kernel::Cx;
use sim_lib_numbers_tensor::{Tensor, TensorLocation};

use crate::{LoweringError, LoweringPlan, PreflightCheck};

pub(crate) struct ResidentExecution {
    pub(crate) real: Tensor,
    pub(crate) imaginary: Tensor,
    pub(crate) uploads: u64,
    pub(crate) submissions: u64,
    pub(crate) segments: u64,
}

pub(crate) fn execute_resident(
    cx: &mut Cx,
    plan: &LoweringPlan,
) -> Result<ResidentExecution, LoweringError> {
    let (mut tiles, uploads) = plan.execute_with_uploaded_inputs(cx)?;
    if tiles.len() != 1 {
        return Err(LoweringError::new(
            PreflightCheck::Execution,
            format!(
                "resident Study requires one admitted output tile, found {}",
                tiles.len()
            ),
        ));
    }
    let lowered = tiles.pop().expect("one resident tile was checked");
    let real = lowered.real().clone();
    let imaginary = lowered.imaginary().clone();
    let real_site = resident_site(&real, "real")?;
    let imaginary_site = resident_site(&imaginary, "imaginary")?;
    if real_site != imaginary_site {
        return Err(LoweringError::new(
            PreflightCheck::Execution,
            format!(
                "resident result components disagree on placement: {real_site} versus {imaginary_site}"
            ),
        ));
    }
    let submissions = lowered
        .submissions()
        .iter()
        .try_fold(0_u64, |total, evidence| {
            total
                .checked_add(u64::try_from(evidence.accepted).map_err(|_| {
                    LoweringError::new(
                        PreflightCheck::Execution,
                        "accepted submission count does not fit u64",
                    )
                })?)
                .ok_or_else(|| {
                    LoweringError::new(
                        PreflightCheck::Execution,
                        "accepted submission count overflowed u64",
                    )
                })
        })?;
    let segments = u64::try_from(lowered.tile().segments_per_tensor())
        .ok()
        .and_then(|segments| segments.checked_mul(2))
        .ok_or_else(|| {
            LoweringError::new(
                PreflightCheck::Execution,
                "final resident segment count overflowed u64",
            )
        })?;
    Ok(ResidentExecution {
        real,
        imaginary,
        uploads: u64::try_from(uploads).map_err(|_| {
            LoweringError::new(
                PreflightCheck::Execution,
                "resident upload count does not fit u64",
            )
        })?,
        submissions,
        segments,
    })
}

fn resident_site(
    tensor: &Tensor,
    component: &'static str,
) -> Result<sim_kernel::Symbol, LoweringError> {
    match tensor.location() {
        TensorLocation::Resident { site, .. } => Ok(site),
        TensorLocation::Host => Err(LoweringError::new(
            PreflightCheck::Execution,
            format!("provider returned host storage for final {component} component"),
        )),
    }
}