Struct TmaStored 

pub struct TmaStored;

A Destination the TMA engine stores: the output is a tensor map, served as a tile a shared-memory box is bulk-stored into (Tile::smem_box, Tile::copy_from), a box at a time, which the descriptor clips at the tensor’s edge. Its width is the descriptor’s cell, so a launch binds it one element wide.

Trait Implementations

impl Destination for TmaStored

type Arg<E: Numeric> = TileArg<'static>

The launch argument carrying the output and its spec, at element E.

fn tile<E: Numeric, V: Size>( arg: &Self::Arg<E>, partitioning: &Partitioning, ) -> Tile

Serve the argument as a Tile under the kernel’s one partitioning, stored in lines of V where the argument does not state its own width.

fn __expand_tile<E: Numeric, V: Size>( scope: &Scope, arg: &<Self::Arg<E> as CubeType>::ExpandType, partitioning: &<Partitioning as CubeType>::ExpandType, ) -> <Tile as CubeType>::ExpandType

impl DestinationLaunch for TmaStored

type Operand = TmaOperand

What the argument is made from, before the kernel’s element types it.

fn arg<E: Numeric>(operand: TmaOperand) -> TileArgLaunch