28 #ifndef __PASO_SPARSEMATRIX_H__ 29 #define __PASO_SPARSEMATRIX_H__ 46 bool patternIsUnrolled);
61 const double* b)
const;
65 SparseMatrix_ptr
unroll(SparseMatrixType type)
const;
69 const index_t* new_col_index)
const;
71 SparseMatrix_ptr
getBlock(
int blockid)
const;
77 void saveMM(
const char* filename)
const;
81 return pattern->borrowMainDiagonalPointer();
86 return pattern->borrowColoringPointer();
91 return pattern->getNumColors();
96 return pattern->maxDeg();
129 static SparseMatrix_ptr
loadMM_toCSR(
const char* filename);
133 const double* mask_col,
134 double main_diagonal_value);
137 const double* mask_col,
138 double main_diagonal_value);
141 double main_diagonal_value);
144 double main_diagonal_value);
147 double main_diagonal_value);
149 void nullifyRows_CSR(
const double* mask_row,
double main_diagonal_value);
181 const_SparseMatrix_ptr A,
183 const double beta,
double* out);
186 const_SparseMatrix_ptr A,
188 const double beta,
double* out);
191 const_SparseMatrix_ptr A,
193 const double beta,
double* out);
196 const_SparseMatrix_ptr A,
198 const double beta,
double* out);
201 const_SparseMatrix_ptr A,
203 const double beta,
double* out);
206 const_SparseMatrix_ptr B);
209 const_SparseMatrix_ptr B,
210 const_SparseMatrix_ptr T);
214 #endif // __PASO_SPARSEMATRIX_H__ void copyBlockToMainDiagonal(const double *in)
Definition: SparseMatrix.cpp:411
void SparseMatrix_MatrixVector_CSC_OFFSET0(const double alpha, const_SparseMatrix_ptr A, const double *in, const double beta, double *out)
Definition: SparseMatrix_MatrixVector.cpp:43
boost::shared_ptr< Pattern > Pattern_ptr
Definition: Pattern.h:36
void SparseMatrix_MatrixVector_CSR_OFFSET1(const double alpha, const_SparseMatrix_ptr A, const double *in, const double beta, double *out)
Definition: SparseMatrix_MatrixVector.cpp:191
dim_t block_size
Definition: SparseMatrix.h:162
SparseMatrixType type
Definition: SparseMatrix.h:159
dim_t getNumColors() const
Definition: SparseMatrix.h:89
boost::shared_ptr< const SparseMatrix > const_SparseMatrix_ptr
Definition: SparseMatrix.h:37
dim_t getTotalNumCols() const
Definition: SparseMatrix.h:104
int SparseMatrixType
Definition: SparseMatrix.h:39
void setValues(double value)
Definition: SparseMatrix.cpp:488
void nullifyRows_CSR(const double *mask_row, double main_diagonal_value)
Definition: SparseMatrix_nullifyRowsAndCols.cpp:142
dim_t numCols
Definition: SparseMatrix.h:164
void maxAbsRow_CSR_OFFSET0(double *array) const
Definition: SparseMatrix.cpp:374
static SparseMatrix_ptr loadMM_toCSR(const char *filename)
Definition: SparseMatrix.cpp:206
void SparseMatrix_MatrixVector_CSR_OFFSET0(const double alpha, const_SparseMatrix_ptr A, const double *in, const double beta, double *out)
Definition: SparseMatrix_MatrixVector.cpp:279
void copyBlockFromMainDiagonal(double *out) const
Definition: SparseMatrix.cpp:423
void * solver_p
pointer to data needed by a solver
Definition: SparseMatrix.h:175
boost::shared_ptr< SparseMatrix > SparseMatrix_ptr
Definition: SparseMatrix.h:35
dim_t row_block_size
Definition: SparseMatrix.h:160
dim_t getNumCols() const
Definition: SparseMatrix.h:114
dim_t col_block_size
Definition: SparseMatrix.h:161
void nullifyRowsAndCols_CSC(const double *mask_row, const double *mask_col, double main_diagonal_value)
Definition: SparseMatrix_nullifyRowsAndCols.cpp:77
void SparseMatrix_MatrixVector_CSC_OFFSET1(const double alpha, const_SparseMatrix_ptr A, const double *in, const double beta, double *out)
Definition: SparseMatrix_MatrixVector.cpp:119
void addRow_CSR_OFFSET0(double *array) const
Definition: SparseMatrix.cpp:393
dim_t numRows
Definition: SparseMatrix.h:163
index_t * borrowColoringPointer() const
Definition: SparseMatrix.h:84
index_t * borrowMainDiagonalPointer() const
Definition: SparseMatrix.h:79
void nullifyRowsAndCols_CSR_BLK1(const double *mask_row, const double *mask_col, double main_diagonal_value)
Definition: SparseMatrix_nullifyRowsAndCols.cpp:59
void addAbsRow_CSR_OFFSET0(double *array) const
Definition: SparseMatrix.cpp:355
SparseMatrix_ptr getTranspose() const
Definition: SparseMatrix.cpp:564
void nullifyRows_CSR_BLK1(const double *mask_row, double main_diagonal_value)
Definition: SparseMatrix_nullifyRowsAndCols.cpp:125
dim_t len
Definition: SparseMatrix.h:166
SparseMatrix_ptr getBlock(int blockid) const
Definition: SparseMatrix_getSubmatrix.cpp:85
void invMain(double *inv_diag, index_t *pivot) const
Definition: SparseMatrix.cpp:503
SparseMatrix_ptr unroll(SparseMatrixType type) const
Definition: SparseMatrix.cpp:640
int index_t
Definition: types.h:24
dim_t maxDeg() const
Definition: SparseMatrix.h:94
void copyFromMainDiagonal(double *out) const
Definition: SparseMatrix.cpp:435
void copyToMainDiagonal(const double *in)
Definition: SparseMatrix.cpp:449
void SparseMatrix_MatrixVector_CSR_OFFSET0_DIAG(const double alpha, const_SparseMatrix_ptr A, const double *in, const double beta, double *out)
Definition: SparseMatrix_MatrixVector.cpp:422
SparseMatrix_ptr getSubmatrix(dim_t n_row_sub, dim_t n_col_sub, const index_t *row_list, const index_t *new_col_index) const
Definition: SparseMatrix_getSubmatrix.cpp:43
Definition: SparseMatrix.h:42
double getSparsity() const
Definition: SparseMatrix.h:124
index_t dim_t
Definition: types.h:27
~SparseMatrix()
Definition: SparseMatrix.cpp:188
SparseMatrix_ptr SparseMatrix_MatrixMatrixTranspose(const_SparseMatrix_ptr A, const_SparseMatrix_ptr B, const_SparseMatrix_ptr T)
Definition: SparseMatrix_MatrixMatrixTranspose.cpp:51
double * val
this is used for classical CSR or CSC
Definition: SparseMatrix.h:169
SparseMatrix_ptr SparseMatrix_MatrixMatrix(const_SparseMatrix_ptr A, const_SparseMatrix_ptr B)
Definition: SparseMatrix_MatrixMatrix.cpp:43
void applyDiagonal_CSR_OFFSET0(const double *left, const double *right)
Definition: SparseMatrix.cpp:463
dim_t getNumRows() const
Definition: SparseMatrix.h:109
index_t solver_package
package controlling the solver pointer
Definition: SparseMatrix.h:172
dim_t getTotalNumRows() const
Definition: SparseMatrix.h:99
void saveHB_CSC(const char *filename) const
Definition: SparseMatrix_saveHB.cpp:179
void nullifyRowsAndCols_CSR(const double *mask_row, const double *mask_col, double main_diagonal_value)
Definition: SparseMatrix_nullifyRowsAndCols.cpp:101
void saveMM(const char *filename) const
Definition: SparseMatrix.cpp:290
SparseMatrix(SparseMatrixType type, Pattern_ptr pattern, dim_t rowBlockSize, dim_t colBlockSize, bool patternIsUnrolled)
Definition: SparseMatrix.cpp:105
double getSize() const
Definition: SparseMatrix.h:119
Pattern_ptr pattern
Definition: SparseMatrix.h:165
void nullifyRowsAndCols_CSC_BLK1(const double *mask_row, const double *mask_col, double main_diagonal_value)
Definition: SparseMatrix_nullifyRowsAndCols.cpp:41
void applyBlockMatrix(double *block_diag, index_t *pivot, double *x, const double *b) const
Definition: SparseMatrix.cpp:555