Enum StageStorage 

pub enum StageStorage {
    Tiled {
        block: Vec<(Axis, usize)>,
        chunks: RowChunks,
    },
    Strided,
    Lines {
        read: UnitRead,
    },
}

Where a stage lives and how it lays its cells out, stated at Stages::smem.

Variants

Tiled

Shared memory grouped into block-sized tiles, one edge per axis of the operation.

Fields

block: Vec<(Axis, usize)>

chunks: RowChunks

Strided

Lines

The plane’s units, one line per unit.

Fields

read: UnitRead

Implementations

impl StageStorage

pub fn tma_swizzle( &self, space: &Space, vector_size: usize, elem_bits: usize, ) -> Result<TensorMapSwizzle, Refusal>

The swizzle a TMA descriptor moving the whole of a stage over space in one box lands its rows with, so they lie as this storage keeps them: the stage’s lines vector_size values of elem_bits bits, which a line holds whole bytes of (a packed operand’s line is its words). What the launch builds the descriptor with, and what the stage fill holds the stage to.

Errors

Blocks splitting a row, which a box lands whole; blocks stacking a number of rows the swizzle’s period of rows does not divide, whose rows a box keys off their place in the stage rather than in the block; padded rows, which a box lands dense, and swizzled rows longer than a descriptor swizzle spans.

pub fn bytes( &self, space: &Space, vector_size: usize, elem_bits: usize, ) -> usize

Bytes a stage over space holds laid out this way, its lines vector_size values of elem_bits bits, which a line holds whole bytes of: every line of its cells, and the padding after each row. What Stages::smem allocates for it.

Trait Implementations

impl Clone for StageStorage

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for StageStorage

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

Formats the value using the given formatter.

impl Eq for StageStorage

impl Hash for StageStorage

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 StageStorage

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for StageStorage