SPV_EXT_cooperative_matrix_maintenance1

OpCooperativeMatrixGetCoordinateEXT

Converts an index in range [0, OpCooperativeMatrixLengthKHR - 1] to a (row,column) coordinate. The element of the matrix at that index (when treated as a composite) is the element located at (row,column) in the matrix. If Index is out of bounds, the resulting value is undefined.

Result Type must be a vector type with two 32-bit integer components.

Matrix must have an OpTypeCooperativeMatrixKHR type.

Index must have a 32-bit integer type.

Capability:
CooperativeMatrixGetCoordinateEXT

5

5363

<id>
Result Type

Result <id>

<id>
Matrix

<id>
Index

OpCooperativeMatrixConvertUseEXT

Converts a cooperative matrix to another cooperative matrix with different Use.

Result Type must be an OpTypeCooperativeMatrixKHR.

The type of Matrix must be an OpTypeCooperativeMatrixKHR with the same Scope as Result Type. The Component Type of Matrix and Result Type must either be the same or be scalar integer types that differ only in Signedness.

If the instruction is not decorated with CooperativeMatrixTransposeEXT, Matrix must have the same Rows and Columns as Result Type.

If the instruction is decorated with CooperativeMatrixTransposeEXT, Matrix must have the same Rows as Result Type has Columns, and vice versa.

If the Use of Matrix is MatrixAccumulatorKHR, then the Use of Result Type must be MatrixAKHR or MatrixBKHR.

If the Use of Matrix is MatrixAKHR or MatrixBKHR, then the Use of Result Type must be MatrixAccumulatorKHR.

For conversions that change both Component Type and Use, use Op{F,S,U,etc.}Convert instead.

Capability:
CooperativeMatrixConversionsEXT

4

5293

<id>
Result Type

Result <id>

<id>
Matrix

OpCooperativeMatrixPerElementOpEXT

Applies an operation to each element of a cooperative matrix.

The type of Matrix must be an OpTypeCooperativeMatrixKHR.

Result Type must match the type of Matrix.

Func must be an OpFunction whose return type matches the component type of Matrix, whose first two parameters must be 32-bit integer types, whose third parameter type must match the component type of Matrix, and which may have additional parameters. The function is called for each element of the matrix where the element is passed as the third parameter to the function, the row and column number of the matrix are passed as the first and second parameters, and any optional operands are passed in order as the remaining parameters.

The number of optional operands must match the number of additional parameters of Func. The type of each optional operand that does not have a cooperative matrix type must match the type of the corresponding parameter of Func.

For each optional operand with a cooperative matrix type, the corresponding matrix element is passed to the function. The matrix type must match the type of Matrix, and the component type of the matrix must match the type of the corresponding parameter of Func.

The return value of Func is the corresponding element of Result.

The calls are considered unordered against each other, and calls may occur more than once.

Within Func and any functions called directly or indirectly by Func, tangled instructions are not allowed.

Capability:
CooperativeMatrixPerElementOperationsEXT

5+variable

5369

<id>
Result Type

Result <id>

<id>
Matrix

<id>
Func

Optional
<id>, <id>, …​

OpCooperativeMatrixReduceEXT

Computes a matrix where each element of the result matrix is computed from a row, column, or neighborhood of the source matrix.

Result Type must be an OpTypeCooperativeMatrixKHR type.

The type of Matrix must be an OpTypeCooperativeMatrixKHR with the same Component Type as Result Type.

The type of Matrix and Result Type must each have Use of MatrixAccumulatorKHR and must have matching Scope.

Reduce must be Row, Column, 2x2, or the combination of Row and Column.

If Reduce includes 2x2, the dimensions of Result Type must be half of the dimensions of Matrix. Each element (r,c) of the result matrix is calculated by combining elements (2*r, 2*c), (2*r+1, 2*c), (2*r, 2*c+1), and (2*r+1, 2*c+1) of Matrix.

If Reduce equals Row, then Result Type must have the same number of rows as Matrix but can have any supported number of columns. All elements of a row in the result matrix have the same value, which is computed by combining all elements of the corresponding row of Matrix.

If Reduce equals Column, then Result Type must have the same number of columns as Matrix but can have any supported number of rows. All elements of a column in the result matrix have the same value, which is computed by combining all elements of the corresponding column of Matrix.

If Reduce includes Row and Column, Result Type can have any number of rows and columns. All elements of the result matrix have the same value, which is computed by combining all elements of Matrix.

CombineFunc must be an OpFunction with two parameters whose types and result type all match the component type of Matrix. This function is called to combine pairs of elements (or intermediate results) when computing the reduction. This function should be mathematically commutative and associative (though in practice, with floating point numbers, may not be exactly commutative/associative).

Within CombineFunc and any functions called directly or indirectly by CombineFunc, tangled instructions are not allowed.

Capability:
CooperativeMatrixReductionsEXT

6

5366

<id>
Result Type

Result <id>

<id>
Matrix

Literal
Reduce

<id>
CombineFunc