Struct CooperativeMatrix 

pub struct CooperativeMatrix<T, const USE: u32, const ROWS: u32, const COLS: u32> { /* private fields */ }

A cooperative matrix distributed across the subgroup.

Each invocation holds a fragment of the full ROWS × COLS matrix. The hardware maps elements to invocations automatically.

Type parameters

• T: element type (f32, f64, i32, u32, i8, u8, etc.)
• USE: matrix role — one of MatrixUse::MatrixA, MatrixUse::MatrixB, MatrixUse::MatrixAccumulator cast to u32
• ROWS: number of rows
• COLS: number of columns

Capability

Requires CooperativeMatrixKHR + SPV_KHR_cooperative_matrix.

Implementations

impl<T, const USE: u32, const ROWS: u32, const COLS: u32> CooperativeMatrix<T, USE, ROWS, COLS>

pub unsafe fn load(slice: &[T], layout: MatrixLayout, stride: u32) -> Self

Load a cooperative matrix through a pointer.

slice must point into an array. layout specifies whether the matrix is stored in row-major (MatrixLayout::RowMajor) or column-major (MatrixLayout::ColumnMajor) order. stride is the number of elements between the start of consecutive rows (row-major) or columns (column-major).

The scope is always Subgroup.

Safety

• slice must point into an array and be valid for all element accesses implied by the matrix dimensions, layout, and stride.
• All operands must be dynamically uniform within every instance of the subgroup scope.

pub unsafe fn store(self, slice: &mut [T], layout: MatrixLayout, stride: u32)

Store a cooperative matrix through a pointer.

slice must point into an array. layout specifies whether the matrix is stored in row-major (MatrixLayout::RowMajor) or column-major (MatrixLayout::ColumnMajor) order. stride is the number of elements between the start of consecutive rows (row-major) or columns (column-major).

The scope is always Subgroup.

Safety

• slice must point into an array and be valid for all element accesses implied by the matrix dimensions, layout, and stride.
• All operands must be dynamically uniform within every instance of the subgroup scope.

pub fn length(&self) -> u32

Returns the number of matrix components this invocation is responsible for.

The sum across all invocations in the subgroup equals ROWS * COLS.

Trait Implementations

impl<T: Clone, const USE: u32, const ROWS: u32, const COLS: u32> Clone for CooperativeMatrix<T, USE, ROWS, COLS>

fn clone(&self) -> CooperativeMatrix<T, USE, ROWS, COLS>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<T: Copy, const USE: u32, const ROWS: u32, const COLS: u32> Copy for CooperativeMatrix<T, USE, ROWS, COLS>