Operator Registry#
The complete operator registry is maintained at FlagSparse conf/operators.yaml.
gather#
Gathers values from a dense tensor by index using the FlagSparse gather path.
Field |
Value |
|---|---|
id |
gather |
kind |
Sparse |
stages |
beta: 1.0 |
labels |
flagsparse, sparse, triton, public-api |
for |
flagsparse_gather |
scatter#
Scatters values into a dense tensor by index using the FlagSparse scatter path.
Field |
Value |
|---|---|
id |
scatter |
kind |
Sparse |
stages |
beta: 1.0 |
labels |
flagsparse, sparse, triton, public-api |
for |
flagsparse_scatter |
spmv_csr#
Computes sparse matrix-vector multiplication for a CSR matrix.
Field |
Value |
|---|---|
id |
spmv_csr |
kind |
SparseLinearAlg |
stages |
beta: 1.0 |
labels |
flagsparse, sparse, csr, triton, public-api |
for |
flagsparse_spmv_csr, prepare_spmv_csr |
spmv_coo#
Computes sparse matrix-vector multiplication for a COO matrix.
Field |
Value |
|---|---|
id |
spmv_coo |
kind |
SparseLinearAlg |
stages |
beta: 1.0 |
labels |
flagsparse, sparse, coo, triton, public-api |
for |
flagsparse_spmv_coo, prepare_spmv_coo |
spmv_coo_tocsr#
Computes COO SpMV through a COO-to-CSR preparation path.
Field |
Value |
|---|---|
id |
spmv_coo_tocsr |
kind |
SparseLinearAlg |
stages |
beta: 1.0 |
labels |
flagsparse, sparse, coo, csr, triton, public-api |
for |
flagsparse_spmv_coo_tocsr, prepare_spmv_coo_tocsr |
spmm_csr#
Computes sparse matrix-dense matrix multiplication for a CSR matrix.
Field |
Value |
|---|---|
id |
spmm_csr |
kind |
SparseLinearAlg |
stages |
beta: 1.0 |
labels |
flagsparse, sparse, csr, triton, public-api |
for |
flagsparse_spmm_csr, flagsparse_spmm_csr_run, prepare_spmm_csr_route |
spmm_coo#
Computes sparse matrix-dense matrix multiplication for a COO matrix.
Field |
Value |
|---|---|
id |
spmm_coo |
kind |
SparseLinearAlg |
stages |
beta: 1.0 |
labels |
flagsparse, sparse, coo, triton, public-api |
for |
flagsparse_spmm_coo |
spmm_csr_opt#
Computes optimized CSR sparse matrix-dense matrix multiplication.
Field |
Value |
|---|---|
id |
spmm_csr_opt |
kind |
SparseLinearAlg |
stages |
beta: 1.0 |
labels |
flagsparse, sparse, csr, triton, optimized, public-api |
for |
flagsparse_spmm_csr_opt, prepare_spmm_csr_opt |
spmm_csr_opt_alg1#
Computes optimized CSR SpMM using the algorithm 1 path.
Field |
Value |
|---|---|
id |
spmm_csr_opt_alg1 |
kind |
SparseLinearAlg |
stages |
beta: 1.0 |
labels |
flagsparse, sparse, csr, triton, optimized, public-api |
for |
flagsparse_spmm_csr_opt_alg1, flagsparse_spmm_csr_opt_alg1_preprocess, prepare_spmm_csr_opt_alg1, prepare_spmm_csr_opt_alg1_preprocess |
spmm_csr_opt_alg2#
Computes optimized CSR SpMM using the algorithm 2 path.
Field |
Value |
|---|---|
id |
spmm_csr_opt_alg2 |
kind |
SparseLinearAlg |
stages |
beta: 1.0 |
labels |
flagsparse, sparse, csr, triton, optimized, public-api |
for |
flagsparse_spmm_csr_opt_alg2, flagsparse_spmm_csr_opt_alg2_preprocess, prepare_spmm_csr_opt_alg2, prepare_spmm_csr_opt_alg2_preprocess |
alpha_spmm_alg1#
Computes AlphaSparse-style CSR SpMM using the algorithm 1 implementation family.
Field |
Value |
|---|---|
id |
alpha_spmm_alg1 |
kind |
SparseLinearAlg |
stages |
beta: 1.0 |
labels |
flagsparse, sparse, csr, triton, alpha-sparse, public-api |
for |
flagsparse_alpha_spmm_alg1, flagsparse_alpha_spmm_alg1_tle, flagsparse_alpha_spmm_alg1_tle_opt, prepare_alpha_spmm_alg1, prepare_alpha_spmm_alg1_tle, prepare_alpha_spmm_alg1_tle_opt |
spgemm_csr#
Computes sparse matrix-sparse matrix multiplication for CSR inputs.
Field |
Value |
|---|---|
id |
spgemm_csr |
kind |
SparseLinearAlg |
stages |
beta: 1.0 |
labels |
flagsparse, sparse, csr, triton, public-api |
for |
flagsparse_spgemm_csr, prepare_spgemm_csr |
sddmm_csr#
Computes sampled dense-dense matrix multiplication on a CSR sparsity pattern.
Field |
Value |
|---|---|
id |
sddmm_csr |
kind |
SparseLinearAlg |
stages |
beta: 1.0 |
labels |
flagsparse, sparse, csr, triton, public-api |
for |
flagsparse_sddmm_csr, prepare_sddmm_csr |
spsv_csr#
Solves sparse triangular systems represented in CSR format.
Field |
Value |
|---|---|
id |
spsv_csr |
kind |
SparseSolver |
stages |
beta: 1.0 |
labels |
flagsparse, sparse, csr, triton, public-api |
for |
flagsparse_spsv_csr |
spsv_coo#
Solves sparse triangular systems represented in COO format.
Field |
Value |
|---|---|
id |
spsv_coo |
kind |
SparseSolver |
stages |
beta: 1.0 |
labels |
flagsparse, sparse, coo, triton, public-api |
for |
flagsparse_spsv_coo |
spsv_descriptor_api#
Provides descriptor, buffer-size, analysis, preprocess, and solve APIs for SpSV workflows.
Field |
Value |
|---|---|
id |
spsv_descriptor_api |
kind |
SparseSolver, Utility |
stages |
beta: 1.0 |
labels |
flagsparse, sparse, csr, coo, descriptor, public-api |
for |
flagsparse_create_spsv_handle, flagsparse_create_dnvec, flagsparse_create_spmat_csr, flagsparse_create_spmat_coo, flagsparse_spsv_buffer_size, flagsparse_spsv_buffer_size_ex, flagsparse_spsv_create_workspace, flagsparse_spsv_analysis_csr, flagsparse_spsv_analysis_coo, flagsparse_spsv_analysis_ex, flagsparse_spsv_preprocess_csr, flagsparse_spsv_preprocess_coo, flagsparse_spsv_solve_csr, flagsparse_spsv_solve_coo, flagsparse_spsv_solve_ex |
spsm_csr#
Solves sparse triangular systems with dense matrix right-hand sides in CSR format.
Field |
Value |
|---|---|
id |
spsm_csr |
kind |
SparseSolver |
stages |
beta: 1.0 |
labels |
flagsparse, sparse, csr, triton, public-api |
for |
flagsparse_spsm_csr |
spsm_coo#
Solves sparse triangular systems with dense matrix right-hand sides in COO format.
Field |
Value |
|---|---|
id |
spsm_coo |
kind |
SparseSolver |
stages |
beta: 1.0 |
labels |
flagsparse, sparse, coo, triton, public-api |
for |
flagsparse_spsm_coo |
sparse_format_constructors#
Creates sparse matrix containers and converts COO data into supported sparse formats.
Field |
Value |
|---|---|
id |
sparse_format_constructors |
kind |
SparseFormat, Utility |
stages |
beta: 1.0 |
labels |
flagsparse, sparse, format, public-api |
for |
create_csr_matrix, create_coo_matrix, create_csc_matrix, create_bsr_matrix, create_sell_matrix, create_blocked_ell_matrix, coo_to_csr, coo_to_csc, coo_to_bsr, coo_to_sell, coo_to_blocked_ell, generate_random_sparse_matrix, read_mtx_file |