Module cooperative_matrix 

Cooperative matrix types and operations (SPV_KHR_cooperative_matrix).

Requires the CooperativeMatrixKHR capability and SPV_KHR_cooperative_matrix extension:

-C target-feature=+CooperativeMatrixKHR,+ext:SPV_KHR_cooperative_matrix

See the SPV_KHR_cooperative_matrix specification for full details.

Structs

CooperativeMatrix

A cooperative matrix distributed across the subgroup.

Enums

MatrixLayout

Memory layout for cooperative matrix load/store operations.

MatrixUse

Matrix role in a cooperative multiply-accumulate operation (D = A × B + C).

Constants

COLUMN_MAJOR

Memory layout: columns are stored contiguously.

MATRIX_A

Matrix role: input operand A in D = A × B + C.

MATRIX_ACCUMULATOR

Matrix role: accumulator / result in D = A × B + C.

MATRIX_B

Matrix role: input operand B in D = A × B + C.

ROW_MAJOR

Memory layout: rows are stored contiguously.

Functions

mul_add

Linear-algebraic matrix multiply of A by B and then component-wise add C.