Module matrix 

Structs

CastOp

Cast a matrix from one type to another. Note: Unlike most matrix ops, this does not have implicit synchronization because there’s no coordination between threads.

ColIndexOp

Attribute(s):

ElementwiseOp

Executes a closure for each element in the matrix. Note: Unlike most matrix ops, this does not have implicit synchronization because there’s no coordination between threads.

FillOp

Fill a matrix with a scalar value. Note: Unlike most matrix ops, this does not have implicit synchronization because there’s no coordination between threads.

LdMatrixOp

Attribute(s):

LoadOp

Attribute(s):

MatrixLayoutAttr

MatrixShapeAttr

MatrixTypeAttr

MmaManualOp

Attribute(s):

MmaManualScaledOp

Attribute(s):

MultiplyAccumulateOp

RowIndexOp

Attribute(s):

StMatrixOp

Attribute(s):

StoreOp

Attribute(s):

Functions

parse_closure

Reusable closure parser for maybe future ops, move this if it’s used elsewhere

print_closure

Reusable closure printer for maybe future ops, move this if it’s used elsewhere