Struct Geometry 

pub struct Geometry {
    pub sublanes: usize,
    pub lanes: usize,
    pub mxu: usize,
}

The hardware tile shape a lattice is padded and laid out to.

Fields

sublanes: usize

Rows per storage tile (the sublane count of a vector register).

lanes: usize

Columns per storage tile (the lane count of a vector register).

mxu: usize

Side length of the square systolic matrix-unit block.

Implementations

impl Geometry

pub const TPU_V: Geometry

The canonical TPU v-class geometry: 8 sublanes, 128 lanes, 128x128 MXU.

pub const TINY: Geometry

A small geometry handy for tests and documentation examples.

pub fn new(sublanes: usize, lanes: usize, mxu: usize) -> Result<Self>

Build a geometry, validating that every dimension is non-zero.

pub const fn tile_len(&self) -> usize

Number of elements in a single storage tile (sublanes * lanes).

pub const fn round_up(n: usize, to: usize) -> usize

Round n up to the next multiple of to.

pub const fn pad_rows(&self, rows: usize) -> usize

Pad a row count up to a whole number of sublane tiles.

pub const fn pad_cols(&self, cols: usize) -> usize

Pad a column count up to a whole number of lane tiles.

pub const fn tile_rows(&self, rows: usize) -> usize

How many storage tiles tall a padded rows is.

pub const fn tile_cols(&self, cols: usize) -> usize

How many storage tiles wide a padded cols is.

pub const fn is_aligned(&self, rows: usize, cols: usize) -> bool

True if both row and column counts already sit on tile boundaries.

Trait Implementations

impl Clone for Geometry

fn clone(&self) -> Geometry

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for Geometry

impl Debug for Geometry

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

Formats the value using the given formatter.

impl Default for Geometry

fn default() -> Self

Returns the “default value” for a type.

impl Eq for Geometry

impl PartialEq for Geometry

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

Tests for self and other values to be equal, and is used by ==.

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

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for Geometry