Struct KdaTensorPlan 

pub struct KdaTensorPlan {
    pub name: &'static str,
    pub kind: KdaShard,
    pub elem_bytes: usize,
    pub full_rows: usize,
    pub full_row_elems: usize,
    pub local_rows: usize,
    pub local_row_elems: usize,
    pub src_row_offset: usize,
    pub src_col_offset: usize,
}

One tensor’s placement. row_elems is the width of a row in elements; for KdaShard::ChannelCols the roles invert and row_elems is the sharded axis.

Fields

name: &'static str

kind: KdaShard

elem_bytes: usize

full_rows: usize

Rows of the full, on-disk tensor.

full_row_elems: usize

Elements per row of the full tensor.

local_rows: usize

Rows this rank keeps.

local_row_elems: usize

Elements per row this rank keeps (differs from full only for ChannelCols).

src_row_offset: usize

Offset, in rows, of this rank’s slice into the full tensor. Always 0 for Replicated and for ChannelCols (which slices columns, not rows).

src_col_offset: usize

Offset, in elements, of this rank’s column slice. Non-zero only for ChannelCols.

Implementations

impl KdaTensorPlan

pub fn local_bytes(&self) -> usize

Bytes this rank stores for this tensor.

pub fn full_bytes(&self) -> usize

Bytes the full tensor occupies on disk.

Trait Implementations

impl Clone for KdaTensorPlan

fn clone(&self) -> KdaTensorPlan

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for KdaTensorPlan

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl PartialEq for KdaTensorPlan

fn eq(&self, other: &KdaTensorPlan) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Eq for KdaTensorPlan

impl StructuralPartialEq for KdaTensorPlan