BlockSparseLayout

class torchao.prototype.dtypes.BlockSparseLayout(blocksize: int = 64)

BlockSparseLayout is a data class that represents the layout of a block sparse matrix.

Variables

blocksize (int) – The size of the blocks in the sparse matrix. Default is 64.