Simplify parameter

This commit is contained in:
Tong Dong Qiu
2022-11-03 16:29:54 +01:00
parent 627c7d50ac
commit d3d9743b81
2 changed files with 6 additions and 6 deletions

View File

@@ -56,7 +56,7 @@ rocalutionSolverBackend<block_size>::~rocalutionSolverBackend() {
template <unsigned int block_size>
void rocalutionSolverBackend<block_size>::initialize(std::shared_ptr<BlockedMatrix> matrix) {
void rocalutionSolverBackend<block_size>::initialize(BlockedMatrix *matrix) {
this->Nb = matrix->Nb;
this->N = Nb * block_size;
this->nnzb = matrix->nnzbs;
@@ -74,7 +74,7 @@ void rocalutionSolverBackend<block_size>::initialize(std::shared_ptr<BlockedMatr
template <unsigned int block_size>
void rocalutionSolverBackend<block_size>::convert_matrix(std::shared_ptr<BlockedMatrix> matrix) {
void rocalutionSolverBackend<block_size>::convert_matrix(BlockedMatrix *matrix) {
Timer t;
for(int i = 0; i < Nb+1; ++i){
@@ -134,14 +134,14 @@ SolverStatus rocalutionSolverBackend<block_size>::solve_system(std::shared_ptr<B
BdaResult &res)
{
if (initialized == false) {
initialize(matrix);
initialize(matrix.get());
}
tmp_rowpointers = new int[Nb+1];
tmp_colindices = new int[nnzb];
tmp_nnzvalues = new double[nnzb*block_size*block_size];
convert_matrix(matrix);
convert_matrix(matrix.get());
rocalution::LocalVector<double> roc_x;
rocalution::LocalVector<double> roc_rhs;

View File

@@ -66,12 +66,12 @@ private:
/// Initialize sizes and allocate memory
/// \param[in] matrix matrix A
void initialize(std::shared_ptr<BlockedMatrix> matrix);
void initialize(BlockedMatrix *matrix);
/// Convert matrix to rocalution format
/// copy matrix to raw pointers, which are given to and freed by rocalution
/// \param[in] matrix matrix A
void convert_matrix(std::shared_ptr<BlockedMatrix> matrix);
void convert_matrix(BlockedMatrix *matrix);
public: