Struct PlacementPlan 

pub struct PlacementPlan {
    pub node_placement: HashMap<usize, Device>,
    pub tensor_placement: HashMap<usize, Device>,
    pub transfer_cost: f64,
}

Device placement plan

Fields

node_placement: HashMap<usize, Device>

Node index -> Device mapping

tensor_placement: HashMap<usize, Device>

Tensor index -> Device mapping

transfer_cost: f64

Estimated transfer cost

Implementations

impl PlacementPlan

pub fn new() -> Self

pub fn single_device( num_nodes: usize, num_tensors: usize, device: Device, ) -> Self

Create a single-device placement plan

pub fn get_node_device(&self, node_idx: usize) -> Option<Device>

Get device for a node

pub fn get_tensor_device(&self, tensor_idx: usize) -> Option<Device>

Get device for a tensor

pub fn count_transfers(&self, graph: &EinsumGraph) -> usize

Count number of cross-device transfers

pub fn devices(&self) -> Vec<Device>

Get list of all devices used in this plan

pub fn summary(&self) -> String

Summary of the placement plan

Trait Implementations

impl Clone for PlacementPlan

fn clone(&self) -> PlacementPlan

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for PlacementPlan

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

Formats the value using the given formatter.

impl Default for PlacementPlan

fn default() -> Self

Returns the “default value” for a type.