spblas_impl.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_SPBLAS_IMPL
#define _H_ALP_SPBLAS_IMPL

#include "blas_sparse_vec.h"

#ifdef __cplusplus
extern "C" {
#endif

#ifndef SPBLAS_PREFIX
 #error "SPBLAS_PREFIX must be defined"
#endif

#define SPBLAS_NAME( name ) SPCONCAT( SPBLAS_PREFIX, name )
#define EXT_SPBLAS_PREFIX SPCONCAT( SPBLAS_PREFIX, ext_ )
#define EXT_SPBLAS_NAME( name ) SPCONCAT( EXT_SPBLAS_PREFIX, name )

#define SPBLAS_RET_T void

SPBLAS_RET_T SPBLAS_NAME( dcsrgemv )(
    const char * transa,
    const int * m,
    const double * a, const int * ia, const int * ja,
    const double * x,
    double * y
);

SPBLAS_RET_T SPBLAS_NAME( dcsrmm )(
    const char * transa,
    const int * m, const int * n, const int * k,
    const double * alpha,
    const char * matdescra, const double * val, const int * indx,
    const int * pntrb, const int * pntre,
    const double * b, const int * ldb,
    const double * beta,
    double * c, const int * ldc
);

SPBLAS_RET_T SPBLAS_NAME( dcsrmultcsr )(
    const char * trans, const int * request, const int * sort,
    const int * m, const int * n, const int * k,
    double * a, int * ja, int * ia,
    double * b, int * jb, int * ib,
    double * c, int * jc, int * ic,
    const int * nzmax, int * info
);

SPBLAS_RET_T EXT_SPBLAS_NAME( dcsrmultsv )(
    const char * trans, const int * request,
    const int * m, const int * n,
    const double * a, const int * ja, const int * ia,
    const extblas_sparse_vector x,
    extblas_sparse_vector y
);

SPBLAS_RET_T EXT_SPBLAS_NAME( free )();
 // ends the SpBLAS doxygen group

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

#endif // end _H_ALP_SPBLAS_IMPL