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_sparse_vector x,
    extblas_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();
 
#ifdef __cplusplus
} // end extern "C"
#endif
 
#endif // end _H_ALP_SPARSEBLAS_NIST