1 #ifndef PARALUTION_GPU_CUSPARSE_CSR_HPP_
2 #define PARALUTION_GPU_CUSPARSE_CSR_HPP_
8 const cusparseMatDescr_t descrA,
int nnzA,
9 const double *csrValA,
const int *csrRowPtrA,
const int *csrColIndA,
11 const cusparseMatDescr_t descrB,
int nnzB,
12 const double *csrValB,
const int *csrRowPtrB,
const int *csrColIndB,
13 const cusparseMatDescr_t descrC,
14 double *csrValC,
int *csrRowPtrC,
int *csrColIndC) {
16 return cusparseDcsrgeam(handle,
20 csrValA, csrRowPtrA, csrColIndA,
23 csrValB, csrRowPtrB, csrColIndB,
25 csrValC, csrRowPtrC, csrColIndC);
30 const cusparseMatDescr_t descrA,
int nnzA,
31 const float *csrValA,
const int *csrRowPtrA,
const int *csrColIndA,
33 const cusparseMatDescr_t descrB,
int nnzB,
34 const float *csrValB,
const int *csrRowPtrB,
const int *csrColIndB,
35 const cusparseMatDescr_t descrC,
36 float *csrValC,
int *csrRowPtrC,
int *csrColIndC) {
37 return cusparseScsrgeam(handle,
41 csrValA, csrRowPtrA, csrColIndA,
44 csrValB, csrRowPtrB, csrColIndB,
46 csrValC, csrRowPtrC, csrColIndC);
51 cusparseOperation_t transA, cusparseOperation_t transB,
53 const cusparseMatDescr_t descrA,
const int nnzA,
54 const double *csrValA,
55 const int *csrRowPtrA,
const int *csrColIndA,
56 const cusparseMatDescr_t descrB,
const int nnzB,
57 const double *csrValB,
58 const int *csrRowPtrB,
const int *csrColIndB,
59 const cusparseMatDescr_t descrC,
61 const int *csrRowPtrC,
int *csrColIndC ) {
63 return cusparseDcsrgemm(handle,
68 csrRowPtrA, csrColIndA,
71 csrRowPtrB, csrColIndB,
74 csrRowPtrC, csrColIndC );
79 cusparseOperation_t transA, cusparseOperation_t transB,
81 const cusparseMatDescr_t descrA,
const int nnzA,
83 const int *csrRowPtrA,
const int *csrColIndA,
84 const cusparseMatDescr_t descrB,
const int nnzB,
86 const int *csrRowPtrB,
const int *csrColIndB,
87 const cusparseMatDescr_t descrC,
89 const int *csrRowPtrC,
int *csrColIndC ) {
91 return cusparseScsrgemm(handle,
96 csrRowPtrA, csrColIndA,
99 csrRowPtrB, csrColIndB,
102 csrRowPtrC, csrColIndC );
cusparseStatus_t __cusparseXcsrgeam__(cusparseHandle_t handle, int m, int n, const double *alpha, const cusparseMatDescr_t descrA, int nnzA, const double *csrValA, const int *csrRowPtrA, const int *csrColIndA, const double *beta, const cusparseMatDescr_t descrB, int nnzB, const double *csrValB, const int *csrRowPtrB, const int *csrColIndB, const cusparseMatDescr_t descrC, double *csrValC, int *csrRowPtrC, int *csrColIndC)
Definition: cusparse_csr.hpp:6
Definition: backend_manager.cpp:43
cusparseStatus_t __cusparseXcsrgemm__(cusparseHandle_t handle, cusparseOperation_t transA, cusparseOperation_t transB, int m, int n, int k, const cusparseMatDescr_t descrA, const int nnzA, const double *csrValA, const int *csrRowPtrA, const int *csrColIndA, const cusparseMatDescr_t descrB, const int nnzB, const double *csrValB, const int *csrRowPtrB, const int *csrColIndB, const cusparseMatDescr_t descrC, double *csrValC, const int *csrRowPtrC, int *csrColIndC)
Definition: cusparse_csr.hpp:50