def triton_sparse_transpose_dense_matmul()

in sparse_autoencoder/kernels.py [0:0]


def triton_sparse_transpose_dense_matmul(
    sparse_indices: torch.Tensor,
    sparse_values: torch.Tensor,
    dense: torch.Tensor,
    N: int,
    BLOCK_SIZE_AK=128,