44 #ifndef AMESOS2_TACHO_DEF_HPP
45 #define AMESOS2_TACHO_DEF_HPP
47 #include <Teuchos_Tuple.hpp>
48 #include <Teuchos_ParameterList.hpp>
49 #include <Teuchos_StandardParameterEntryValidators.hpp>
52 #include "Amesos2_Tacho_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 )
69 template <
class Matrix,
class Vector>
73 data_.solver.release();
77 template <
class Matrix,
class Vector>
81 std::ostringstream oss;
82 oss <<
"Tacho solver interface";
86 template<
class Matrix,
class Vector>
93 template <
class Matrix,
class Vector>
97 #ifdef HAVE_AMESOS2_TIMERS
98 Teuchos::TimeMonitor symFactTime( this->timers_.symFactTime_ );
103 if(do_optimization()) {
104 this->matrixA_->returnRowPtr_kokkos_view(host_row_ptr_view_);
105 this->matrixA_->returnColInd_kokkos_view(host_cols_view_);
108 data_.solver.setSolutionMethod(data_.method);
109 data_.solver.setLevelSetOptionAlgorithmVariant(data_.variant);
115 data_.solver.analyze(this->globalNumCols_, host_row_ptr_view_, host_cols_view_);
116 data_.solver.initialize();
122 template <
class Matrix,
class Vector>
126 #ifdef HAVE_AMESOS2_TIMERS
127 Teuchos::TimeMonitor numFactTimer(this->timers_.numFactTime_);
132 if(do_optimization()) {
133 this->matrixA_->returnValues_kokkos_view(device_nzvals_view_);
135 data_.solver.factorize(device_nzvals_view_);
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();
157 #ifdef HAVE_AMESOS2_TIMERS
158 Teuchos::TimeMonitor mvConvTimer(this->timers_.vecConvTime_);
159 Teuchos::TimeMonitor redistTimer(this->timers_.vecRedistTime_);
161 const bool initialize_data =
true;
162 const bool do_not_initialize_data =
false;
163 Util::get_1d_copy_helper_kokkos_view<MultiVecAdapter<Vector>,
164 device_solve_array_t>::do_get(initialize_data, B, this->bValues_,
166 ROOTED, this->rowIndexBase_);
167 bDidAssignX = Util::get_1d_copy_helper_kokkos_view<MultiVecAdapter<Vector>,
168 device_solve_array_t>::do_get(do_not_initialize_data, X, this->xValues_,
170 ROOTED, this->rowIndexBase_);
176 #ifdef HAVE_AMESOS2_TIMER
177 Teuchos::TimeMonitor solveTimer(this->timers_.solveTime_);
180 if (workspace_.extent(0) < this->globalNumRows_ || workspace_.extent(1) < nrhs) {
181 workspace_ = device_solve_array_t(
182 Kokkos::ViewAllocateWithoutInitializing(
"t"), this->globalNumRows_, nrhs);
185 data_.solver.solve(xValues_, bValues_, workspace_);
194 Teuchos::broadcast(*(this->getComm()), 0, &ierr);
196 TEUCHOS_TEST_FOR_EXCEPTION( ierr != 0, std::runtime_error,
197 "tacho_solve has error code: " << ierr );
204 #ifdef HAVE_AMESOS2_TIMERS
205 Teuchos::TimeMonitor redistTimer(this->timers_.vecRedistTime_);
210 Util::template put_1d_data_helper_kokkos_view<
213 ROOTED, this->rowIndexBase_);
220 template <
class Matrix,
class Vector>
225 return( this->matrixA_->getGlobalNumRows() == this->matrixA_->getGlobalNumCols() );
229 template <
class Matrix,
class Vector>
233 RCP<const Teuchos::ParameterList> valid_params = getValidParameters_impl();
238 auto method_name = parameterList->get<std::string> (
"method",
"chol");
239 if (method_name ==
"chol")
241 else if (method_name ==
"ldl")
243 else if (method_name ==
"lu")
246 std::cout <<
"Error: not supported solution method\n";
249 data_.variant = parameterList->get<
int> (
"variant", 2);
256 template <
class Matrix,
class Vector>
257 Teuchos::RCP<const Teuchos::ParameterList>
260 static Teuchos::RCP<const Teuchos::ParameterList> valid_params;
262 if( is_null(valid_params) ){
263 Teuchos::RCP<Teuchos::ParameterList> pl = Teuchos::parameterList();
265 pl->set(
"method",
"chol",
"Type of factorization, chol, ldl, or lu");
266 pl->set(
"variant", 2,
"Type of solver variant, 0, 1, or 2");
278 template <
class Matrix,
class Vector>
281 return (this->root_ && (this->matrixA_->getComm()->getSize() == 1));
284 template <
class Matrix,
class Vector>
289 if(current_phase == SOLVE) {
293 if(!do_optimization()) {
294 #ifdef HAVE_AMESOS2_TIMERS
295 Teuchos::TimeMonitor convTimer(this->timers_.mtxConvTime_);
305 device_nzvals_view_ = device_value_type_array(
306 Kokkos::ViewAllocateWithoutInitializing(
"nzvals"), this->globalNumNonZeros_);
307 host_cols_view_ = host_ordinal_type_array(
308 Kokkos::ViewAllocateWithoutInitializing(
"colind"), this->globalNumNonZeros_);
309 host_row_ptr_view_ = host_size_type_array(
310 Kokkos::ViewAllocateWithoutInitializing(
"rowptr"), this->globalNumRows_ + 1);
313 typename host_size_type_array::value_type nnz_ret = 0;
315 #ifdef HAVE_AMESOS2_TIMERS
316 Teuchos::TimeMonitor mtxRedistTimer( this->timers_.mtxRedistTime_ );
319 TEUCHOS_TEST_FOR_EXCEPTION( this->rowIndexBase_ != this->columnIndexBase_,
321 "Row and column maps have different indexbase ");
324 device_value_type_array, host_ordinal_type_array, host_size_type_array>::do_get(
325 this->matrixA_.ptr(),
331 this->columnIndexBase_);
339 template<
class Matrix,
class Vector>
345 #endif // AMESOS2_TACHO_DEF_HPP
Amesos2::SolverCore: A templated interface for interaction with third-party direct sparse solvers...
Definition: Amesos2_SolverCore_decl.hpp:105
EPhase
Used to indicate a phase in the direct solution.
Definition: Amesos2_TypeDecl.hpp:65
Amesos2 interface to the Tacho package.
Definition: Amesos2_Tacho_decl.hpp:67
int symbolicFactorization_impl()
Perform symbolic factorization of the matrix using Tacho.
Definition: Amesos2_Tacho_def.hpp:95
Utility functions for Amesos2.
bool matrixShapeOK_impl() const
Determines whether the shape of the matrix is OK for this solver.
Definition: Amesos2_Tacho_def.hpp:222
int numericFactorization_impl()
Tacho specific numeric factorization.
Definition: Amesos2_Tacho_def.hpp:124
int preOrdering_impl()
Performs pre-ordering on the matrix to increase efficiency.
Definition: Amesos2_Tacho_def.hpp:88
Teuchos::RCP< const Teuchos::ParameterList > getValidParameters_impl() const
Definition: Amesos2_Tacho_def.hpp:258
Similar to get_ccs_helper , but used to get a CRS representation of the given matrix.
Definition: Amesos2_Util.hpp:659
std::string description() const override
Returns a short description of this Solver.
Definition: Amesos2_Tacho_def.hpp:79
int solve_impl(const Teuchos::Ptr< MultiVecAdapter< Vector > > X, const Teuchos::Ptr< const MultiVecAdapter< Vector > > B) const
Tacho specific solve.
Definition: Amesos2_Tacho_def.hpp:142
TachoSolver(Teuchos::RCP< const Matrix > A, Teuchos::RCP< Vector > X, Teuchos::RCP< const Vector > B)
Initialize from Teuchos::RCP.
Definition: Amesos2_Tacho_def.hpp:58
bool loadA_impl(EPhase current_phase)
Reads matrix data into internal structures.
Definition: Amesos2_Tacho_def.hpp:286
bool do_optimization() const
can we optimize size_type and ordinal_type for straight pass through
Definition: Amesos2_Tacho_def.hpp:280
A templated MultiVector class adapter for Amesos2.
Definition: Amesos2_MultiVecAdapter_decl.hpp:176
~TachoSolver()
Destructor.
Definition: Amesos2_Tacho_def.hpp:70