nvmath.bindings.cublasLt.MatrixLayoutAttribute

enum nvmath.bindings.cublasLt.MatrixLayoutAttribute(value)[source]

See cublasLtMatrixLayoutAttribute_t.

Member Type:

int

Valid values are as follows:

TYPE = <MatrixLayoutAttribute.TYPE: 0>
ORDER = <MatrixLayoutAttribute.ORDER: 1>
ROWS = <MatrixLayoutAttribute.ROWS: 2>
COLS = <MatrixLayoutAttribute.COLS: 3>
LD = <MatrixLayoutAttribute.LD: 4>
BATCH_COUNT = <MatrixLayoutAttribute.BATCH_COUNT: 5>
STRIDED_BATCH_OFFSET = <MatrixLayoutAttribute.STRIDED_BATCH_OFFSET: 6>
PLANE_OFFSET = <MatrixLayoutAttribute.PLANE_OFFSET: 7>