Struct TpManifest 

pub struct TpManifest {
    pub chips: usize,
    pub chip_of: BTreeMap<String, u16>,
    pub slices: BTreeMap<String, TpSlice>,
    pub collectives: Vec<TpCollective>,
}

How a tensor-parallel program is spread over chips RDUs.

Fields

chips: usize

chip_of: BTreeMap<String, u16>

Every chip-local tensor, by name. A tensor not named here is replicated: the spine every chip shares.

slices: BTreeMap<String, TpSlice>

Every chip-local tensor that is a slice of a full checkpoint (or carried-state) tensor.

collectives: Vec<TpCollective>

Implementations

impl TpManifest

pub fn new(chips: usize) -> Self

pub fn place(&mut self, name: &str, chip: usize, slice: Option<TpSlice>)

Record name as chip chip’s tensor, a slice of slice.of when given.

pub fn collective_folding(&self, binder: &str) -> Option<&TpCollective>

The collective whose fold statements include binder, if any.

pub fn chip_running(&self, name: &str) -> u16

The chip a statement binding name runs on: its tensor’s chip, or the root (chip 0) for the replicated spine.

Trait Implementations

impl Clone for TpManifest

fn clone(&self) -> TpManifest

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TpManifest

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

Formats the value using the given formatter.

impl Default for TpManifest

fn default() -> TpManifest

Returns the “default value” for a type.

impl PartialEq for TpManifest

fn eq(&self, other: &TpManifest) -> 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 Serialize for TpManifest

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl Eq for TpManifest

impl StructuralPartialEq for TpManifest