Struct DistributedPlacementPlan 

pub struct DistributedPlacementPlan {
    pub node_placement: HashMap<usize, Device>,
    pub sharding_specs: Vec<ShardingSpec>,
    pub communication_deps: HashMap<usize, Vec<usize>>,
}

Placement plan for distributed execution.

Fields

node_placement: HashMap<usize, Device>

Node to device mapping

sharding_specs: Vec<ShardingSpec>

Sharding specifications for model parallelism

communication_deps: HashMap<usize, Vec<usize>>

Communication dependencies (node -> nodes it depends on)

Implementations

impl DistributedPlacementPlan

pub fn new() -> Self

Create a new empty placement plan.

pub fn place_node(&mut self, node_id: usize, device: Device)

Add a node placement.

pub fn add_sharding(&mut self, spec: ShardingSpec)

Add a sharding specification.

pub fn get_device(&self, node_id: usize) -> Option<&Device>

Get the device for a node.

pub fn get_sharding(&self, node_id: usize) -> Option<&ShardingSpec>

Get sharding spec for a node.

pub fn is_sharded(&self, node_id: usize) -> bool

Check if a node is sharded.

Trait Implementations

impl Clone for DistributedPlacementPlan

fn clone(&self) -> DistributedPlacementPlan

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DistributedPlacementPlan

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

Formats the value using the given formatter.

impl Default for DistributedPlacementPlan

fn default() -> Self

Returns the “default value” for a type.