cusparse_interface.h

Go to the documentation of this file.

/*
 *  SPDX-FileCopyrightText: Copyright 2021, Siavash Ameli <sameli@berkeley.edu>
 *  SPDX-License-Identifier: BSD-3-Clause
 *  SPDX-FileType: SOURCE
 *
 *  This program is free software: you can redistribute it and/or modify it
 *  under the terms of the license found in the LICENSE.txt file in the root
 *  directory of this source tree.
 */
 
#ifndef _CU_BASIC_ALGEBRA_CUSPARSE_INTERFACE_H_
#define _CU_BASIC_ALGEBRA_CUSPARSE_INTERFACE_H_
 
 
// =======
// Headers
// =======
 
#include <cusparse.h>  // cusparseSpMatDescr_t, cusparseDnVecDescr_t,
                       // CusparseStatus_t, CUSPARSE_STATUS_SUCCESS,
                       // cusparseCreateCsr, cusparseCreateDnVec,
                       // cusparseDestroySpMat, cusparseDestroyDnVec,
                       // CUDA_R_32F, CUDA_R_64F, CUSPARSE_INDEX_32I,
                       // CUSPARSE_INDEX_BASE_ZERO, cusparseHandle_t,
                       // cusparseSpMVAlg_t, cusparseSpMV_buffer_size
#include "../_definitions/types.h"  // LongIndexType
 
 
// ==================
// cusparse interface
// ==================
 
 
namespace cusparse_interface
{
    // create cusparse matrix
    template <typename DataType>
    void create_cusparse_matrix(
        cusparseSpMatDescr_t& cusparse_matrix,
        const LongIndexType num_rows,
        const LongIndexType num_columns,
        const LongIndexType nnz,
        DataType* device_A_data,
        LongIndexType* device_A_indices,
        LongIndexType* device_A_index_pointer);
 
    // create cusparse vector
    template <typename DataType>
    void create_cusparse_vector(
        cusparseDnVecDescr_t& cusparse_vector,
        const LongIndexType vector_size,
        DataType* device_vector);
 
    // destroy cusparse matrix
    void destroy_cusparse_matrix(
        cusparseSpMatDescr_t& cusparse_matrix);
 
    // destroy cusparse vector
    void destroy_cusparse_vector(
        cusparseDnVecDescr_t& cusparse_vector);
 
    // cusparse matrix buffer size
    template <typename DataType>
    void cusparse_matrix_buffer_size(
            cusparseHandle_t cusparse_handle,
            cusparseOperation_t cusparse_operation,
            const DataType alpha,
            cusparseSpMatDescr_t cusparse_matrix,
            cusparseDnVecDescr_t cusparse_input_vector,
            const DataType beta,
            cusparseDnVecDescr_t cusparse_output_vector,
            cusparseSpMVAlg_t algorithm,
            size_t* buffer_size);
 
    // cusparse matvec
    template <typename DataType>
    void cusparse_matvec(
            cusparseHandle_t cusparse_handle,
            cusparseOperation_t cusparse_operation,
            const DataType alpha,
            cusparseSpMatDescr_t cusparse_matrix,
            cusparseDnVecDescr_t cusparse_input_vector,
            const DataType beta,
            cusparseDnVecDescr_t cusparse_output_vector,
            cusparseSpMVAlg_t algorithm,
            void* external_buffer);
}  // namespace cusparse_interface
 
 
#endif  //  _CU_BASIC_ALGEBRA_CUSPARSE_INTERFACE_H_