Struct cubecl_core::frontend::cmma::Matrix

pub struct Matrix<C: CubeType> { /* private fields */ }

A matrix represent a 2D grid of numbers.

They can either be in a row major or a column major format.

Implementations

impl<C: CubePrimitive> Matrix<C>

pub fn new( ident: MatrixIdent, m: u32, n: u32, k: u32, layout: MatrixLayout, ) -> Self

Create a new matrix that is going to be used in the matrix-multiply and accumulate function.

You have to declare the shape used for the execution. The shape of the current matrix is determined using the MatrixIdent.

• MatrixIdent::A Shape => (M, K)
• MatrixIdent::B Shape => (K, N)
• MatrixIdent::Accumulator Shape => (M, N)

Not all shapes are supported, and the permitted shapes depend on the element type.

Refer to nvidia documentation.

pub fn __expand_new( context: &mut CubeContext, ident: MatrixIdent, m: ExpandElementTyped<UInt>, n: ExpandElementTyped<UInt>, k: ExpandElementTyped<UInt>, layout: MatrixLayout, ) -> MatrixExpand

Trait Implementations

impl<C: Clone + CubeType> Clone for Matrix<C>

fn clone(&self) -> Matrix<C>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<C: CubeType> CubeType for Matrix<C>

type ExpandType = MatrixExpand

fn init(context: &mut CubeContext, expand: Self::ExpandType) -> Self::ExpandType

Wrapper around the init method, necesary to type inference.

impl<C: Copy + CubeType> Copy for Matrix<C>