44 #ifndef AMESOS2_UMFPACK_DEF_HPP
45 #define AMESOS2_UMFPACK_DEF_HPP
47 #include <Teuchos_Tuple.hpp>
48 #include <Teuchos_ParameterList.hpp>
49 #include <Teuchos_StandardParameterEntryValidators.hpp>
52 #include "Amesos2_Umfpack_decl.hpp"
57 template <
class Matrix,
class Vector>
59 Teuchos::RCP<const Matrix> A,
60 Teuchos::RCP<Vector> X,
61 Teuchos::RCP<const Vector> B )
63 , is_contiguous_(true)
65 data_.Symbolic = NULL;
70 template <
class Matrix,
class Vector>
73 if (data_.Symbolic) function_map::umfpack_free_symbolic (&data_.Symbolic);
74 if (data_.Numeric) function_map::umfpack_free_numeric (&data_.Numeric);
77 template <
class Matrix,
class Vector>
81 std::ostringstream oss;
82 oss <<
"Umfpack solver interface";
86 template<
class Matrix,
class Vector>
94 template <
class Matrix,
class Vector>
100 if (data_.Symbolic) {
101 function_map::umfpack_free_symbolic(&(data_.Symbolic));
104 function_map::umfpack_defaults(data_.Control);
106 status = function_map::umfpack_symbolic(
107 this->globalNumRows_,this->globalNumCols_,
108 &(this->colptr_view_[0]), &(this->rowind_view_[0]),
109 &(this->nzvals_view_[0]), &(data_.Symbolic), data_.Control, data_.Info);
116 template <
class Matrix,
class Vector>
122 if(!data_.Symbolic) {
123 symbolicFactorization_impl();
126 function_map::umfpack_defaults(data_.Control);
129 function_map::umfpack_free_numeric(&(data_.Numeric));
132 status = function_map::umfpack_numeric(
133 &(this->colptr_view_[0]),
134 &(this->rowind_view_[0]), &(this->nzvals_view_[0]), data_.Symbolic,
135 &(data_.Numeric), data_.Control, data_.Info);
140 template <
class Matrix,
class Vector>
147 const global_size_type ld_rhs = this->root_ ? X->getGlobalLength() : 0;
148 const size_t nrhs = X->getGlobalNumVectors();
150 const size_t val_store_size = as<size_t>(ld_rhs * nrhs);
151 Teuchos::Array<umfpack_type> xValues(val_store_size);
152 Teuchos::Array<umfpack_type> bValues(val_store_size);
155 #ifdef HAVE_AMESOS2_TIMERS
156 Teuchos::TimeMonitor mvConvTimer(this->timers_.vecConvTime_);
157 Teuchos::TimeMonitor redistTimer( this->timers_.vecRedistTime_ );
161 (is_contiguous_ ==
true) ? ROOTED : CONTIGUOUS_AND_ROOTED,
162 this->rowIndexBase_);
165 int UmfpackRequest = this->control_.useTranspose_ ? UMFPACK_At : UMFPACK_A;
171 #ifdef HAVE_AMESOS2_TIMER
172 Teuchos::TimeMonitor solveTimer(this->timers_.solveTime_);
174 if (data_.Symbolic) {
175 function_map::umfpack_free_symbolic(&(data_.Symbolic));
179 int i_ld_rhs = as<int>(ld_rhs);
181 for(
size_t j = 0 ; j < nrhs; j++) {
182 int status = function_map::umfpack_solve(
184 &(this->colptr_view_[0]), &(this->rowind_view_[0]), &(this->nzvals_view_[0]),
185 &xValues.getRawPtr()[j*i_ld_rhs],
186 &bValues.getRawPtr()[j*i_ld_rhs],
187 data_.Numeric, data_.Control, data_.Info);
198 Teuchos::broadcast(*(this->getComm()), 0, &ierr);
200 TEUCHOS_TEST_FOR_EXCEPTION( ierr != 0, std::runtime_error,
201 "umfpack_solve has error code: " << ierr );
205 #ifdef HAVE_AMESOS2_TIMERS
206 Teuchos::TimeMonitor redistTimer(this->timers_.vecRedistTime_);
211 (is_contiguous_ ==
true) ? ROOTED : CONTIGUOUS_AND_ROOTED,
212 this->rowIndexBase_);
219 template <
class Matrix,
class Vector>
226 return( this->matrixA_->getGlobalNumRows() == this->matrixA_->getGlobalNumCols() );
230 template <
class Matrix,
class Vector>
235 using Teuchos::getIntegralValue;
236 using Teuchos::ParameterEntryValidator;
238 RCP<const Teuchos::ParameterList> valid_params = getValidParameters_impl();
240 if( parameterList->isParameter(
"IsContiguous") ){
241 is_contiguous_ = parameterList->get<
bool>(
"IsContiguous");
246 template <
class Matrix,
class Vector>
247 Teuchos::RCP<const Teuchos::ParameterList>
250 static Teuchos::RCP<const Teuchos::ParameterList> valid_params;
252 if( is_null(valid_params) ){
253 Teuchos::RCP<Teuchos::ParameterList> pl = Teuchos::parameterList();
255 pl->set(
"IsContiguous",
true,
"Whether GIDs contiguous");
264 template <
class Matrix,
class Vector>
268 if(current_phase == SOLVE) {
272 #ifdef HAVE_AMESOS2_TIMERS
273 Teuchos::TimeMonitor convTimer(this->timers_.mtxConvTime_);
278 Kokkos::resize(nzvals_view_,this->globalNumNonZeros_);
279 Kokkos::resize(rowind_view_,this->globalNumNonZeros_);
280 Kokkos::resize(colptr_view_,this->globalNumCols_ + 1);
285 #ifdef HAVE_AMESOS2_TIMERS
286 Teuchos::TimeMonitor mtxRedistTimer( this->timers_.mtxRedistTime_ );
289 TEUCHOS_TEST_FOR_EXCEPTION( this->rowIndexBase_ != this->columnIndexBase_,
291 "Row and column maps have different indexbase ");
293 host_size_type_array>::do_get(this->matrixA_.ptr(),
294 nzvals_view_, rowind_view_, colptr_view_, nnz_ret,
295 (is_contiguous_ ==
true) ? ROOTED : CONTIGUOUS_AND_ROOTED,
297 this->rowIndexBase_);
304 template<
class Matrix,
class Vector>
310 #endif // AMESOS2_UMFPACK_DEF_HPP
Amesos2::SolverCore: A templated interface for interaction with third-party direct sparse solvers...
Definition: Amesos2_SolverCore_decl.hpp:105
bool matrixShapeOK_impl() const
Determines whether the shape of the matrix is OK for this solver.
Definition: Amesos2_Umfpack_def.hpp:221
A generic helper class for getting a CCS representation of a Matrix.
Definition: Amesos2_Util.hpp:648
Teuchos::RCP< const Teuchos::ParameterList > getValidParameters_impl() const
Definition: Amesos2_Umfpack_def.hpp:248
EPhase
Used to indicate a phase in the direct solution.
Definition: Amesos2_TypeDecl.hpp:65
int symbolicFactorization_impl()
Perform symbolic factorization of the matrix using Umfpack.
Definition: Amesos2_Umfpack_def.hpp:96
Umfpack(Teuchos::RCP< const Matrix > A, Teuchos::RCP< Vector > X, Teuchos::RCP< const Vector > B)
Initialize from Teuchos::RCP.
Definition: Amesos2_Umfpack_def.hpp:58
int solve_impl(const Teuchos::Ptr< MultiVecAdapter< Vector > > X, const Teuchos::Ptr< const MultiVecAdapter< Vector > > B) const
Umfpack specific solve.
Definition: Amesos2_Umfpack_def.hpp:142
Helper class for getting 1-D copies of multivectors.
Definition: Amesos2_MultiVecAdapter_decl.hpp:267
Utility functions for Amesos2.
bool loadA_impl(EPhase current_phase)
Reads matrix data into internal structures.
Definition: Amesos2_Umfpack_def.hpp:266
int numericFactorization_impl()
Umfpack specific numeric factorization.
Definition: Amesos2_Umfpack_def.hpp:118
Amesos2 interface to the Umfpack package.
Definition: Amesos2_Umfpack_decl.hpp:63
~Umfpack()
Destructor.
Definition: Amesos2_Umfpack_def.hpp:71
int preOrdering_impl()
Performs pre-ordering on the matrix to increase efficiency.
Definition: Amesos2_Umfpack_def.hpp:88
std::string description() const
Returns a short description of this Solver.
Definition: Amesos2_Umfpack_def.hpp:79
Helper class for putting 1-D data arrays into multivectors.
Definition: Amesos2_MultiVecAdapter_decl.hpp:373
A templated MultiVector class adapter for Amesos2.
Definition: Amesos2_MultiVecAdapter_decl.hpp:176