Enum PlaneFlowPartitionExpand 

pub enum PlaneFlowPartitionExpand {
    MainFlowOnly,
    LoadOnlyFirst(<PartitionThreshold as CubeType>::ExpandType),
    LoadOnlyLast(<PartitionThreshold as CubeType>::ExpandType),
}

Variants

MainFlowOnly

LoadOnlyFirst(<PartitionThreshold as CubeType>::ExpandType)

LoadOnlyLast(<PartitionThreshold as CubeType>::ExpandType)

Implementations

impl PlaneFlowPartitionExpand

pub fn __expand_compute_index_method( self, scope: &Scope, ) -> <u32 as CubeType>::ExpandType

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

pub fn __expand_load_index_method( self, scope: &Scope, 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_method( &self, scope: &Scope, ) -> <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_method( self, scope: &Scope, ) -> <bool as CubeType>::ExpandType

Whether the current plane is a load-only plane.

pub fn __expand_is_compute_plane_method( self, scope: &Scope, ) -> <bool as CubeType>::ExpandType

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, ) -> <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, 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, ) -> <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, ) -> <bool as CubeType>::ExpandType

Whether the current plane is a load-only plane.

pub fn __expand_is_compute_plane( scope: &Scope, this: Self, ) -> <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 AsMutExpand for PlaneFlowPartitionExpand

fn __expand_ref_mut_method(&mut self, _: &Scope) -> &mut Self

fn __expand_as_mut_method(&mut self, scope: &Scope) -> &mut T

impl AsRefExpand for PlaneFlowPartitionExpand

fn __expand_ref_method(&self, _: &Scope) -> &Self

fn __expand_as_ref_method(&self, scope: &Scope) -> &T

impl CubeDebug for PlaneFlowPartitionExpand

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 CubeEnum for PlaneFlowPartitionExpand

type RuntimeValue = ()

fn discriminant(&self) -> NativeExpand<i32>

fn runtime_value(self) -> Self::RuntimeValue

Return the runtime value of this enum, if only one variant has a value. Should return () for all other cases.

fn discriminant_of(variant_name: &'static str) -> i32

fn discriminant_of_value(&self, variant_name: &'static str) -> i32

impl ExpandTypeClone for PlaneFlowPartitionExpand

fn clone_unchecked(&self) -> Self

Unchecked clone that only clones the conceptual runtime value. Should only be used in cases where each copy is used in a mutually exclusive branch (i.e. match, runtime enums). This is intentionally separated from Rust’s Clone semantics and should only be used for the conceptual expand values, never real data. Using two values in the same branch is undefined behaviour.

impl IntoExpand for PlaneFlowPartitionExpand

type Expand = PlaneFlowPartitionExpand

fn into_expand(self, _: &Scope) -> Self

impl IntoMut for PlaneFlowPartitionExpand

fn into_mut(self, scope: &Scope) -> Self

Convert the variable into a potentially new mutable variable in scope, copying if needed.