Skip to content

dsdbsparse#

Includes our CUDA general datastructure kernels.

Functions:

  • find_ranks –

    Finds the ranks of the indices in the offsets.

find_ranks #

find_ranks(nnz_section_offsets: NDArray, inds: NDArray) -> NDArray

Finds the ranks of the indices in the offsets.

Parameters:

  • nnz_section_offsets (NDArray) –

    The offsets of the non-zero sections.

  • inds (NDArray) –

    The indices to find the ranks for.

Returns:

  • ranks ( NDArray ) –

    The ranks of the indices in the offsets.