sim-lib-compute-wgpu 0.1.0

Evidence-based wgpu tensor compute site discovery for SIM.
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

//! Transfer planning for segmented wgpu tensor payloads.

use crate::WgpuSegmentPlan;

/// One host-to-device or device-to-host copy span.
#[derive(Clone, Debug, PartialEq, Eq)]
pub struct WgpuTransferSpan {
    /// Byte offset from the logical tensor start.
    pub offset: u64,
    /// Copy length in bytes.
    pub bytes: u64,
}

/// Planned transfer spans for a segment plan.
#[derive(Clone, Debug, PartialEq, Eq)]
pub struct WgpuTransferPlan {
    /// Copy spans.
    pub spans: Vec<WgpuTransferSpan>,
}

impl WgpuTransferPlan {
    /// Builds a transfer plan aligned to resident segments.
    pub fn from_segments(plan: &WgpuSegmentPlan) -> Self {
        Self {
            spans: plan
                .segments
                .iter()
                .map(|segment| WgpuTransferSpan {
                    offset: segment.offset,
                    bytes: segment.bytes,
                })
                .collect(),
        }
    }
}