Enum MemberSharding 

pub enum MemberSharding {
    Replicated,
    Equal {
        axis: usize,
    },
    Balanced {
        axis: usize,
    },
    Partitioned {
        axis: usize,
    },
    PartitionedSegments {
        axis: usize,
        segments: Vec<Range<usize>>,
    },
    Segmented {
        axis: usize,
        segments: Vec<Range<usize>>,
    },
}

Rank-local selection rule for one physical checkpoint tensor.

Variants

Replicated

Materialize the complete member on every tensor-parallel rank.

Equal

Split an axis into equal contiguous shards.

Fields

axis: usize

Source tensor axis to partition.

Balanced

Split an axis into balanced, potentially uneven contiguous ranges.

Fields

axis: usize

Source tensor axis to partition.

Partitioned

Map the group’s logical partition onto one physical tensor axis.

Fields

axis: usize

Source tensor axis to partition.

PartitionedSegments

Map the same group-level logical range into each supplied source segment.

Fields

axis: usize

Source tensor axis containing the fused segments.

segments: Vec<Range<usize>>

Ordered, non-overlapping physical source ranges.

Segmented

Partition each supplied source range independently.

Fields

axis: usize

Source tensor axis containing the fused segments.

segments: Vec<Range<usize>>

Ordered, non-overlapping source ranges.

Trait Implementations

impl Clone for MemberSharding

fn clone(&self) -> MemberSharding

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MemberSharding

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

Formats the value using the given formatter.

impl Eq for MemberSharding

impl PartialEq for MemberSharding

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for MemberSharding