Enum PlaneFlowPartition 

pub enum PlaneFlowPartition {
    MainFlowOnly,
    LoadOnlyFirst(PartitionThreshold),
    LoadOnlyLast(PartitionThreshold),
}

Runtime view of PlaneFlowPartitionRule: distinguishes a plane’s role based on its plane id.

Variants

MainFlowOnly

All planes are in the main flow (no specialization).

LoadOnlyFirst(PartitionThreshold)

Load-only planes: [0, Threshold); main-flow planes: [Threshold, total).

LoadOnlyLast(PartitionThreshold)

Main-flow planes: [0, Threshold); load-only planes: [Threshold, total).

Implementations

impl PlaneFlowPartition

pub fn new_MainFlowOnly() -> Self

pub fn __expand_new_MainFlowOnly(_: &Scope) -> PlaneFlowPartitionExpand

pub fn new_LoadOnlyFirst(_0: PartitionThreshold) -> Self

pub fn __expand_new_LoadOnlyFirst( _: &Scope, _0: <PartitionThreshold as CubeType>::ExpandType, ) -> PlaneFlowPartitionExpand

pub fn new_LoadOnlyLast(_0: PartitionThreshold) -> Self

pub fn __expand_new_LoadOnlyLast( _: &Scope, _0: <PartitionThreshold as CubeType>::ExpandType, ) -> PlaneFlowPartitionExpand

impl PlaneFlowPartition

pub fn new(comptime_rule: PlaneFlowPartitionRule) -> PlaneFlowPartition

Construct from comptime rule.

pub fn compute_index(self) -> u32

The index of the current plane among planes that perform compute, ignoring load-only planes.

pub fn load_index(self, specialization_tensor_config: InputLoadFlow) -> u32

The index of the current plane among planes that perform loading, ignoring any plane that does not participate for this ident.

pub fn elect_load_leader(&self) -> bool

Whether this unit is the leader of the loading units. Always the lowest unit in the correct group. Used by TMA; plane_broadcast / plane_elect keep the value warp-uniform.

pub fn is_load_plane(self) -> bool

Whether the current plane is a load-only plane.

pub fn is_compute_plane(self) -> bool

Whether this plane is part of the compute planes. Used in specialized kernels; plane_broadcast keeps the value warp-uniform.

pub fn __expand_new( scope: &Scope, comptime_rule: PlaneFlowPartitionRule, ) -> <PlaneFlowPartition as CubeType>::ExpandType

Construct from comptime rule.

pub fn __expand_compute_index( scope: &Scope, this: <Self as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

The index of the current plane among planes that perform compute, ignoring load-only planes.

pub fn __expand_load_index( scope: &Scope, this: <Self as CubeType>::ExpandType, specialization_tensor_config: InputLoadFlow, ) -> <u32 as CubeType>::ExpandType

The index of the current plane among planes that perform loading, ignoring any plane that does not participate for this ident.

pub fn __expand_elect_load_leader( scope: &Scope, this: &<Self as CubeType>::ExpandType, ) -> <bool as CubeType>::ExpandType

Whether this unit is the leader of the loading units. Always the lowest unit in the correct group. Used by TMA; plane_broadcast / plane_elect keep the value warp-uniform.

pub fn __expand_is_load_plane( scope: &Scope, this: <Self as CubeType>::ExpandType, ) -> <bool as CubeType>::ExpandType

Whether the current plane is a load-only plane.

pub fn __expand_is_compute_plane( scope: &Scope, this: <Self as CubeType>::ExpandType, ) -> <bool as CubeType>::ExpandType

Whether this plane is part of the compute planes. Used in specialized kernels; plane_broadcast keeps the value warp-uniform.

Trait Implementations

impl Clone for PlaneFlowPartition

fn clone(&self) -> PlaneFlowPartition

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for PlaneFlowPartition

impl CubeDebug for PlaneFlowPartition

fn set_debug_name(&self, scope: &Scope, name: &'static str)

Set the debug name of this type’s expansion. Should do nothing for types that don’t appear at runtime

impl CubeType for PlaneFlowPartition

type ExpandType = PlaneFlowPartitionExpand

impl Debug for PlaneFlowPartition

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

Formats the value using the given formatter.

impl Eq for PlaneFlowPartition

impl Hash for PlaneFlowPartition

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for PlaneFlowPartition

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for PlaneFlowPartition