3 #ifndef DUNE_ISTL_SPQR_HH
4 #define DUNE_ISTL_SPQR_HH
6 #if HAVE_SUITESPARSE_SPQR || defined DOXYGEN
11 #include <SuiteSparseQR.hpp>
13 #include <dune/common/exceptions.hh>
14 #include <dune/common/unused.hh>
34 template<
class M,
class T,
class TM,
class TD,
class TA>
35 class SeqOverlappingSchwarz;
37 template<
class T,
bool tag>
38 struct SeqOverlappingSchwarzAssemblerHelper;
45 template<
class Matrix>
62 template<
typename T,
typename A,
int n,
int m>
64 :
public InverseOperator<BlockVector<FieldVector<T,m>, typename A::template rebind<FieldVector<T,m> >::other>,
65 BlockVector<FieldVector<T,n>, typename A::template rebind<FieldVector<T,n> >::other> >
83 return SolverCategory::Category::sequential;
94 SPQR(
const Matrix& matrix,
int verbose=0) : matrixIsLoaded_(false), verbose_(verbose)
97 static_assert((std::is_same<T,double>::value) || (std::is_same<T,std::complex<double> >::value),
98 "Unsupported Type in SPQR (only double and std::complex<double> supported)");
99 cc_ =
new cholmod_common();
100 cholmod_l_start(cc_);
112 SPQR(
const Matrix& matrix,
int verbose,
bool) : matrixIsLoaded_(false), verbose_(verbose)
115 static_assert((std::is_same<T,double>::value) || (std::is_same<T,std::complex<double> >::value),
116 "Unsupported Type in SPQR (only double and std::complex<double> supported)");
117 cc_ =
new cholmod_common();
118 cholmod_l_start(cc_);
123 SPQR() : matrixIsLoaded_(false), verbose_(0)
126 static_assert((std::is_same<T,double>::value) || (std::is_same<T,std::complex<double> >::value),
127 "Unsupported Type in SPQR (only double and std::complex<double> supported)");
128 cc_ =
new cholmod_common();
129 cholmod_l_start(cc_);
135 if ((spqrMatrix_.N() + spqrMatrix_.M() > 0) || matrixIsLoaded_)
137 cholmod_l_finish(cc_);
143 const std::size_t dimMat(spqrMatrix_.N());
145 for(std::size_t k = 0; k != dimMat; ++k)
146 (
static_cast<T*
>(B_->x))[k] = b[k];
147 cholmod_dense* BTemp = B_;
148 B_ = SuiteSparseQR_qmult<T>(0, spqrfactorization_, B_, cc_);
149 cholmod_dense* X = SuiteSparseQR_solve<T>(1, spqrfactorization_, B_, cc_);
150 cholmod_l_free_dense(&BTemp, cc_);
152 for(std::size_t k = 0; k != dimMat; ++k)
153 x [k] = (
static_cast<T*
>(X->x))[k];
154 cholmod_l_free_dense(&X, cc_);
160 std::cout<<std::endl<<
"Solving with SuiteSparseQR"<<std::endl;
161 std::cout<<
"Flops Taken: "<<cc_->SPQR_flopcount<<std::endl;
162 std::cout<<
"Analysis Time: "<<cc_->SPQR_analyze_time<<
" s"<<std::endl;
163 std::cout<<
"Factorize Time: "<<cc_->SPQR_factorize_time<<
" s"<<std::endl;
164 std::cout<<
"Backsolve Time: "<<cc_->SPQR_solve_time<<
" s"<<std::endl;
165 std::cout<<
"Peak Memory Usage: "<<cc_->memory_usage<<
" bytes"<<std::endl;
166 std::cout<<
"Rank Estimate: "<<cc_->SPQR_istat[4]<<std::endl<<std::endl;
173 DUNE_UNUSED_PARAMETER(reduction);
179 DUNE_UNUSED_PARAMETER(option);
180 DUNE_UNUSED_PARAMETER(value);
186 if ((spqrMatrix_.N() + spqrMatrix_.M() > 0) || matrixIsLoaded_)
188 spqrMatrix_ = matrix;
195 if ((spqrMatrix_.N() + spqrMatrix_.M() > 0) || matrixIsLoaded_)
197 spqrMatrix_.setMatrix(matrix,rowIndexSet);
216 return spqrfactorization_;
234 cholmod_l_free_sparse(&A_, cc_);
235 cholmod_l_free_dense(&B_, cc_);
236 SuiteSparseQR_free<T>(&spqrfactorization_, cc_);
238 matrixIsLoaded_ =
false;
248 template<
class M,
class X,
class TM,
class TD,
class T1>
256 const std::size_t dimMat(spqrMatrix_.N());
257 const std::size_t nnz(spqrMatrix_.getColStart()[dimMat]);
259 A_ = cholmod_l_allocate_sparse(dimMat, dimMat, nnz, 1, 1, 0, 1, cc_);
261 for(std::size_t k = 0; k != (dimMat+1); ++k)
262 (
static_cast<long int *
>(A_->p))[k] = spqrMatrix_.getColStart()[k];
263 for(std::size_t k = 0; k != nnz; ++k)
265 (
static_cast<long int*
>(A_->i))[k] = spqrMatrix_.getRowIndex()[k];
266 (
static_cast<T*
>(A_->x))[k] = spqrMatrix_.getValues()[k];
269 B_ = cholmod_l_allocate_dense(dimMat, 1, dimMat, A_->xtype, cc_);
271 spqrfactorization_=SuiteSparseQR_factorize<T>(SPQR_ORDERING_DEFAULT,SPQR_DEFAULT_TOL,A_,cc_);
274 SPQRMatrix spqrMatrix_;
275 bool matrixIsLoaded_;
280 SuiteSparseQR_factorization<T>* spqrfactorization_;
283 template<
typename T,
typename A>
289 template<
typename T,
typename A>
298 template<
typename TL,
typename M>
299 std::shared_ptr<Dune::InverseOperator<typename Dune::TypeListElement<1, TL>::type,
300 typename Dune::TypeListElement<2, TL>::type>>
303 isValidBlock<
typename Dune::TypeListElement<1, TL>::type::block_type>::value,
int> = 0)
const
305 int verbose = config.get(
"verbose", 0);
306 return std::make_shared<Dune::SPQR<M>>(
mat,verbose);
310 template<
typename TL,
typename M>
311 std::shared_ptr<Dune::InverseOperator<typename Dune::TypeListElement<1, TL>::type,
312 typename Dune::TypeListElement<2, TL>::type>>
314 std::enable_if_t<!
isValidBlock<
typename Dune::TypeListElement<1, TL>::type::block_type>::value,
int> = 0)
const
317 "Unsupported Type in SPQR (only double and std::complex<double> supported)");
328 #endif //HAVE_SUITESPARSE_SPQR
329 #endif //DUNE_ISTL_SPQR_HH