def triton_coo_sparse_dense_matmul()

in sparse_autoencoder/kernels.py [0:0]


def triton_coo_sparse_dense_matmul(
    coo_indices: torch.Tensor,
    coo_values: torch.Tensor,
    dense: torch.Tensor,
    N: int,
    BLOCK_SIZE_AK=128,