Struct StoragePartitioning 

pub struct StoragePartitioning { /* private fields */ }

How a tensor’s values are laid down: stated tiles, finest first, then the rest’s order.

// 32 x 32 tiles of a [k, n] weight, stored for a four-wide read, the tiles along N
let storage = StorageLevels::new(&[(N, 4)])
    .tile(&[(N, 8), (K, 32)])
    .grid(&[N, K]);

Implementations

impl StoragePartitioning

pub fn new(geometry: &Geometry, labels: &[Axis]) -> Option<Self>

The statement a bound buffer was written from, read off its dims by stride; labels name the trailing dims. None where the buffer’s order is no partitioning.

pub fn tiles(&self) -> &[(Axis, usize)]

The stated tiles, finest first.

pub fn order(&self) -> &[Axis]

The order the rest of each axis follows, finest first.

pub fn labels(&self, axes: &[Axis]) -> Vec<Axis>

The buffer’s dims, coarsest first, each named by the axis it is a piece of.

pub fn holds( &self, tile: &[(Axis, usize)], extents: &[(Axis, usize)], ) -> Result<(), TileMisfit>

Whether tile, finest first, is held by this partitioning over a tensor of extents. Stated tiles are taken whole; the rest of an axis may be divided.

pub fn physical( &self, extents: &[(Axis, usize)], ) -> Result<Geometry, StorageMisfit>

The buffer storing a tensor of extents: its dims in labels’ order and its Tiling.

Trait Implementations

impl Clone for StoragePartitioning

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 StoragePartitioning

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

Formats the value using the given formatter.

impl Eq for StoragePartitioning

impl Hash for StoragePartitioning

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 StoragePartitioning

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for StoragePartitioning