blas_sparse.h

Go to the documentation of this file.

/*
 *   Copyright 2021 Huawei Technologies Co., Ltd.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef _H_ALP_SPARSEBLAS_NIST
#define _H_ALP_SPARSEBLAS_NIST

#include "blas_sparse_vec.h"

#ifdef __cplusplus
extern "C" {
#endif

enum blas_trans_type {
    blas_no_trans = 0,
    blas_trans,
    blas_conj_trans
};

enum blas_order_type {
    blas_rowmajor,
    blas_colmajor
};

typedef void * blas_sparse_matrix;

blas_sparse_matrix BLAS_duscr_begin( const int m, const int n );

int BLAS_duscr_insert_entry(
    blas_sparse_matrix A,
    const double val,
    const int row, const int col
);

int BLAS_duscr_insert_entries(
    blas_sparse_matrix A,
    const int nnz,
    const double * vals, const int * rows, const int * cols
);

int BLAS_duscr_insert_col(
    blas_sparse_matrix A,
    const int j, const int nnz,
    const double * vals, const int * rows
);

int BLAS_duscr_insert_row(
    blas_sparse_matrix A,
    const int i, const int nnz,
    const double * vals, const int * cols
);

int BLAS_duscr_end( blas_sparse_matrix A );

int BLAS_usds( blas_sparse_matrix A );

int BLAS_dusmv(
    const enum blas_trans_type transa,
    const double alpha, const blas_sparse_matrix A,
    const double * const x, int incx,
    double * const y, const int incy
);

int BLAS_dusmm(
    const enum blas_order_type order,
    const enum blas_trans_type transa,
    const int nrhs,
    const double alpha, const blas_sparse_matrix A,
    const double * B, const int ldb,
    const double * C, const int ldc
);

int EXTBLAS_dusmsv(
    const enum blas_trans_type transa,
    const double alpha, const blas_sparse_matrix A,
    const EXTBLAS_TYPE( sparse_vector ) x,
    EXTBLAS_TYPE( sparse_vector ) y
);

int EXTBLAS_dusmsm(
    const enum blas_trans_type transa,
    const double alpha, const blas_sparse_matrix A,
    const enum blas_trans_type transb, const blas_sparse_matrix B,
    blas_sparse_matrix C
);

int EXTBLAS_dusm_nz( const blas_sparse_matrix A, int * nz );

int EXTBLAS_dusm_open( const blas_sparse_matrix A );

int EXTBLAS_dusm_get(
    const blas_sparse_matrix A,
    double * value, int * row, int * col
);

int EXTBLAS_dusm_close( const blas_sparse_matrix A );

int EXTBLAS_dusm_clear( blas_sparse_matrix A );

int EXTBLAS_free();
 // ends the SparseBLAS doxygen group

#ifdef __cplusplus
} // end extern "C"
#endif

#endif // end _H_ALP_SPARSEBLAS_NIST