tilelang.utils.sparse¶

Attributes¶

compress_util

Functions¶

compress_sm90(A, block_k, transposed)

Module Contents¶

tilelang.utils.sparse.compress_util¶
tilelang.utils.sparse.compress_sm90(A, block_k, transposed)¶
Parameters:

  • A (torch.Tensor)

  • block_k (int)

  • transposed (bool)

Return type:

tuple[torch.Tensor, torch.Tensor]