Shortcuts

BlockSparseLayout

class torchao.dtypes.BlockSparseLayout(blocksize: int = 64)[source]

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.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources