iterative-solver 0.0
propose_rspace.h
1#ifndef LINEARALGEBRA_SRC_MOLPRO_LINALG_ITSOLV_PROPOSE_RSPACE_H
2#define LINEARALGEBRA_SRC_MOLPRO_LINALG_ITSOLV_PROPOSE_RSPACE_H
3#include <molpro/linalg/itsolv/IterativeSolver.h>
4#include <molpro/linalg/itsolv/helper.h>
5#include <molpro/linalg/itsolv/qspace_options.h>
6#include <molpro/linalg/itsolv/rspace_options.h>
7#include <molpro/linalg/itsolv/subspace/Dimensions.h>
8#include <molpro/linalg/itsolv/subspace/ISubspaceSolver.h>
9#include <molpro/linalg/itsolv/subspace/IXSpace.h>
10#include <molpro/linalg/itsolv/subspace/QSpace.h>
11#include <molpro/linalg/itsolv/subspace/gram_schmidt.h>
12#include <molpro/linalg/itsolv/subspace/util.h>
13#include <molpro/linalg/itsolv/util.h>
14#include <molpro/linalg/itsolv/wrap_util.h>
15#include <molpro/profiler/Profiler.h>
16
17#include <algorithm>
18#include <cassert>
19#include <cstddef>
20#include <format>
21#include <functional>
22#include <ranges>
23#include <utility>
24#include <vector>
25
27
28template <class R>
29void normalise(VecRef<R>& params, array::ArrayHandler<R, R>& handler, Logger& logger,
32 for (auto& p : params) {
33 auto dot = handler.dot(p, p);
34 dot = std::sqrt(std::abs(dot));
35 if (dot > thresh) {
36 handler.scal(1. / dot, p);
37 } else {
38 logger.warn("parameter's length is too small for normalisation, dot = " + std::format("{:.2e}", double(dot)));
39 }
40 }
41}
42namespace dspace {
43
52template <typename value_type>
54 const std::vector<std::size_t>& remove_qspace, Logger& logger) {
55 logger.trace("construct_projected_solution()");
56 const auto nQd = remove_qspace.size();
57 const auto nSol = solutions.rows();
58 auto solutions_proj = subspace::Matrix<value_type>({nSol, nQd + dims.nD});
59 for (size_t i = 0; i < nSol; ++i) {
60 for (size_t j = 0; j < nQd; ++j) {
61 solutions_proj(i, j) = solutions(i, dims.oQ + remove_qspace[j]);
62 }
63 for (size_t j = 0; j < dims.nD; ++j) {
64 solutions_proj(i, nQd + j) = solutions(i, dims.oD + j);
65 }
66 }
67 logger.debug("nSol, nQd, nD", nSol, nQd, dims.nD);
68 return solutions_proj;
69}
70
85template <typename value_type>
87 const subspace::Matrix<value_type>& overlap,
88 const subspace::Dimensions& dims,
89 const std::vector<std::size_t>& remove_qspace, Logger& logger) {
90 logger.trace("construct_projected_solution_overlap()");
91 const auto nSol = solutions_proj.rows();
92 const auto nQd = remove_qspace.size();
93 auto overlap_proj = subspace::Matrix<value_type>({nSol, nSol});
94 for (size_t i = 0; i < nSol; ++i) {
95 for (size_t ii = 0; ii <= i; ++ii) {
96 for (size_t j = 0; j < nQd; ++j) {
97 for (size_t k = 0; k < nQd; ++k) {
98 overlap_proj(i, ii) += solutions_proj(i, j) * solutions_proj(ii, k) *
99 overlap(dims.oQ + remove_qspace[j], dims.oQ + remove_qspace[k]);
100 }
101 for (size_t k = 0; k < dims.nD; ++k) {
102 overlap_proj(i, ii) +=
103 solutions_proj(i, j) * solutions_proj(ii, nQd + k) * overlap(dims.oQ + remove_qspace[j], dims.oD + k);
104 }
105 }
106 for (size_t j = 0; j < dims.nD; ++j) {
107 for (size_t k = 0; k < dims.nD; ++k) {
108 overlap_proj(i, ii) +=
109 solutions_proj(i, nQd + j) * solutions_proj(ii, nQd + k) * overlap(dims.oD + j, dims.oD + k);
110 }
111 for (size_t k = 0; k < nQd; ++k) {
112 overlap_proj(i, ii) +=
113 solutions_proj(i, nQd + j) * solutions_proj(ii, k) * overlap(dims.oD + j, dims.oQ + remove_qspace[k]);
114 }
115 }
116 overlap_proj(ii, i) = overlap_proj(i, ii);
117 }
118 }
119 return overlap_proj;
120}
121
129template <typename value_type, typename value_type_abs>
131 const value_type_abs norm_thresh, Logger& logger) {
132 logger.trace("remove_null_norm_and_normalise()");
133 const auto nSol = parameters.rows();
134 auto norm_proj = std::vector<value_type_abs>(nSol, 0.);
135 for (size_t i = 0; i < nSol; ++i)
136 norm_proj[i] = std::sqrt(std::abs(overlap(i, i)));
137 for (size_t i = 0, j = 0; i < nSol; ++i) {
138 if (norm_proj[i] > norm_thresh) {
139 parameters.row(j).scal(1. / norm_proj[i]);
140 overlap.col(j).scal(1. / norm_proj[i]);
141 overlap.row(j).scal(1. / norm_proj[i]);
142 ++j;
143 } else {
144 parameters.remove_row(j);
145 overlap.remove_row_col(j, j);
146 std::stringstream ss;
147 ss << std::setprecision(3) << "remove projected solution parameter i = " << i << ", norm = " << norm_proj[i];
148 logger.info(ss.str());
149 }
150 }
151 logger.data_dump("norm = ", norm_proj);
152 logger.data_dump("parameters after normalisation = ", parameters);
153 logger.data_dump("overlap of parameters = ", overlap);
154}
155
168template <typename value_type, typename value_type_abs>
170 const subspace::Matrix<value_type>& overlap_proj, const value_type_abs svd_thresh,
171 Logger& logger) {
172 logger.trace("remove_null_projected_solutions()");
173 logger.debug("nS on entry = ", solutions_proj.rows());
174 value_type* m = const_cast<std::vector<value_type>&>(overlap_proj.data()).data();
175 auto svd_vecs = svd_system(overlap_proj.rows(), overlap_proj.cols(), array::Span(m, overlap_proj.size()),
176 std::numeric_limits<value_type_abs>::max(), true);
177 svd_vecs.remove_if([&svd_thresh](const auto& el) { return el.value < svd_thresh; });
178 svd_vecs.sort([](const auto& lt, const auto& rt) { return lt.value < rt.value; });
179 const auto nD = svd_vecs.size();
180 const auto nX = solutions_proj.cols();
181 auto solutions_stable = subspace::Matrix<value_type>({nD, nX});
182 auto svd = svd_vecs.begin();
183 for (size_t i = 0; i < nD; ++i, ++svd)
184 for (size_t j = 0; j < overlap_proj.cols(); ++j)
185 for (size_t k = 0; k < nX; ++k)
186 solutions_stable(i, k) += svd->v[j] * solutions_proj(j, k);
187 logger.debug("nS without null space = ", solutions_stable.rows());
188 return solutions_stable;
189}
190
200template <typename value_type>
202 const subspace::Dimensions& dims, const std::vector<std::size_t>& remove_qspace,
203 const subspace::Matrix<value_type>& overlap, const size_t nR) {
204 const auto nDnew = solutions_proj.rows();
205 const auto nQd = remove_qspace.size();
206 const auto nQ = dims.nQ - nQd;
207 auto ov = overlap;
208 for (size_t i = 0; i < dims.nD; ++i) {
209 ov.remove_row_col(dims.oD, dims.oD);
210 }
211 auto is_Qdelete = [&remove_qspace](size_t i) {
212 return std::find(begin(remove_qspace), end(remove_qspace), i) != end(remove_qspace);
213 };
214 for (size_t i = 0, j = 0; i < dims.nQ; ++i) {
215 if (is_Qdelete(i))
216 ov.remove_row_col(dims.oQ + j, dims.oQ + j);
217 else
218 ++j;
219 }
220 const auto oDnew = dims.nP + nQ + nR;
221 ov.resize({oDnew + nDnew, oDnew + nDnew});
222 /*
223 * append overlap with solutions_proj
224 * x_i = \sum_j c_ij v_j
225 * <x_i, v_j> = \sum_k c_ik <v_k, v_j>
226 * <x_i, x_j> = \sum_kl c_ik c_jl <v_k, v_l>
227 */
228 auto accumulate_ov_offdiag = [&](size_t i, size_t j, size_t jj) {
229 for (size_t k = 0; k < nQd; ++k)
230 ov(oDnew + i, j) += solutions_proj(i, k) * overlap(jj, dims.oQ + remove_qspace[k]);
231 for (size_t k = 0; k < dims.nD; ++k)
232 ov(oDnew + i, j) += solutions_proj(i, nQd + k) * overlap(jj, dims.oD + k);
233 ov(j, oDnew + i) = ov(oDnew + i, j);
234 };
235 for (size_t i = 0; i < nDnew; ++i) {
236 for (size_t j = 0; j < dims.nP; ++j)
237 accumulate_ov_offdiag(i, j, dims.oP + j);
238 for (size_t j = 0, jj = 0; j < dims.nQ; ++j)
239 if (!is_Qdelete(j))
240 accumulate_ov_offdiag(i, dims.nP + jj++, dims.oQ + j);
241 for (size_t j = 0; j < nR; ++j)
242 accumulate_ov_offdiag(i, dims.nP + nQ + j, dims.nX + j);
243 }
244 for (size_t i = 0; i < nDnew; ++i) {
245 for (size_t j = 0; j <= i; ++j) {
246 for (size_t k = 0; k < nQd; ++k) {
247 for (size_t l = 0; l < nQd; ++l)
248 ov(oDnew + i, oDnew + j) += solutions_proj(i, k) * solutions_proj(j, l) *
249 overlap(dims.oQ + remove_qspace[k], dims.oQ + remove_qspace[l]);
250 for (size_t l = 0; l < dims.nD; ++l)
251 ov(oDnew + i, oDnew + j) +=
252 solutions_proj(i, k) * solutions_proj(j, nQd + l) * overlap(dims.oQ + remove_qspace[k], dims.oD + l);
253 }
254 for (size_t k = 0; k < dims.nD; ++k) {
255 for (size_t l = 0; l < nQd; ++l)
256 ov(oDnew + i, oDnew + j) +=
257 solutions_proj(i, nQd + k) * solutions_proj(j, l) * overlap(dims.oD + k, dims.oQ + remove_qspace[l]);
258 for (size_t l = 0; l < dims.nD; ++l)
259 ov(oDnew + i, oDnew + j) +=
260 solutions_proj(i, nQd + k) * solutions_proj(j, nQd + l) * overlap(dims.oD + k, dims.oD + l);
261 }
262 ov(oDnew + j, oDnew + i) = ov(oDnew + i, oDnew + j);
263 }
264 }
265 return ov;
266}
267} // namespace dspace
268
281template <class R, class Q, class P, typename value_type>
283 const CVecRef<P>& pparams, const CVecRef<Q>& qparams, const CVecRef<Q>& dparams,
284 ArrayHandlers<R, Q, P>& handlers, Logger& logger) {
285 logger.trace("append_overlap_with_r()");
286 const auto nP = pparams.size(), nQ = qparams.size(), nD = dparams.size(), nN = params.size();
287 const auto nX = nP + nQ + nD + nN;
288 const auto oP = 0;
289 const auto oQ = oP + nP;
290 const auto oD = oQ + nQ;
291 const auto oN = oD + nD;
292 auto ov = overlap;
293 ov.resize({nX, nX}); // solutions are last
294 ov.slice({oN, oN}, {oN + nN, oN + nN}) = subspace::util::overlap(params, handlers.rr());
295 ov.slice({oN, oP}, {oN + nN, oP + nP}) = subspace::util::overlap(params, pparams, handlers.rp());
296 ov.slice({oN, oQ}, {oN + nN, oQ + nQ}) = subspace::util::overlap(params, qparams, handlers.rq());
297 ov.slice({oN, oD}, {oN + nN, oD + nD}) = subspace::util::overlap(params, dparams, handlers.rq());
298 auto copy_upper_to_lower = [&ov, oN, nN](size_t oX, size_t nX) {
299 for (size_t i = 0; i < nX; ++i)
300 for (size_t j = 0; j < nN; ++j)
301 ov(oX + i, oN + j) = ov(oN + j, oX + i);
302 };
303 copy_upper_to_lower(oP, nP);
304 copy_upper_to_lower(oQ, nQ);
305 copy_upper_to_lower(oD, nD);
306 logger.data_dump("full overlap P+Q+D+R = ", ov);
307 return ov;
308}
309
318template <typename value_type>
319std::vector<std::size_t> limit_qspace_size(const subspace::Dimensions& dims, const QSpaceOptions& opts,
320 const subspace::Matrix<value_type>& solutions, Logger& logger) {
321 using value_type_abs = decltype(std::abs(solutions(0, 0)));
322 using contrib_pair = std::pair<std::size_t, value_type_abs>;
323 using std::begin;
324 using std::end;
325
326 logger.trace("limit_qspace_size()");
327
328 assert(opts.max_size >= opts.min_size);
329
330 const bool must_trim_size = dims.nQ > opts.max_size;
331 const bool may_trim_size = dims.nQ > opts.min_size && opts.contrib_thresh > 0;
332
333 if (!must_trim_size && !may_trim_size) {
334 return {};
335 }
336
337 const std::size_t min_deletions = must_trim_size ? dims.nQ - opts.max_size : 0;
338 const std::size_t max_deletions = dims.nQ - opts.min_size;
339
340 std::vector<std::size_t> q_delete;
341 q_delete.reserve(min_deletions);
342
343 const std::size_t nSol = solutions.rows();
344
345 std::vector<contrib_pair> max_contrib_to_solution;
346 max_contrib_to_solution.reserve(nSol);
347
348 for (std::size_t idx : std::ranges::views::iota(std::size_t(0), dims.nQ)) {
349 max_contrib_to_solution.emplace_back(std::make_pair(idx, value_type_abs(0)));
350
351 for (size_t j = 0; j < nSol; ++j) {
352 const value_type_abs current = std::abs(solutions(j, dims.oQ + idx));
353 max_contrib_to_solution.back().second = std::max(max_contrib_to_solution.back().second, current);
354 }
355 }
356
357 std::ranges::sort(max_contrib_to_solution, std::less<>{}, &contrib_pair::second);
358
359 logger.trace("contribution to solutions =",
360 max_contrib_to_solution | std::ranges::views::transform([](const contrib_pair& p) { return p.second; }));
361
362 for (auto [idx, contrib] : max_contrib_to_solution) {
363 if (q_delete.size() == max_deletions || (q_delete.size() >= min_deletions && contrib >= opts.contrib_thresh)) {
364 break;
365 }
366
367 if (opts.contrib_thresh != 0 && contrib >= opts.contrib_thresh) {
368 logger.warn("deleted Q vector with max. contribution = ", contrib);
369 } else {
370 logger.debug("deleted Q vector with max. contribution = ", contrib);
371 }
372
373 q_delete.emplace_back(idx);
374 }
375
376 return q_delete;
377}
378
390template <class R, class Q, class P, typename value_type, typename value_type_abs>
392 const std::vector<std::size_t>& q_delete, const value_type_abs norm_thresh,
393 const value_type_abs svd_thresh, array::ArrayHandler<Q, Q>& handler, Logger& logger) {
394 const auto dims = xspace.dimensions();
395 const auto overlap = xspace.data.at(subspace::EqnData::S);
396 auto solutions_proj = dspace::construct_projected_solution(solutions, dims, q_delete, logger);
397 auto overlap_proj = dspace::construct_projected_solutions_overlap(solutions_proj, overlap, dims, q_delete, logger);
398 dspace::remove_null_norm_and_normalise(solutions_proj, overlap_proj, norm_thresh, logger);
399 solutions_proj = dspace::remove_null_projected_solutions(solutions_proj, overlap_proj, svd_thresh, logger);
400 overlap_proj = dspace::construct_projected_solutions_overlap(solutions_proj, overlap, dims, q_delete, logger);
401 dspace::remove_null_norm_and_normalise(solutions_proj, overlap_proj, norm_thresh, logger);
402 auto overlap_full_subspace = dspace::construct_full_subspace_overlap(solutions_proj, dims, q_delete, overlap, 0);
403 auto svd_vecs = svd_system(overlap_full_subspace.rows(), overlap_full_subspace.cols(),
404 array::Span(&overlap_full_subspace(0, 0), overlap_full_subspace.size()), svd_thresh, true);
405 assert(svd_vecs.empty() && "P+Q+D subspace should be stable by construction");
406 const auto nD = solutions_proj.rows();
407 const auto nQd = q_delete.size();
408 assert(nQd + dims.nD == solutions_proj.cols());
409 const auto &qparams = xspace.cparamsq(), qactions = xspace.cactionsq();
410 const auto &dparams = xspace.cparamsd(), dactions = xspace.cactionsd();
411 std::vector<Q> dparams_new, dactions_new;
412 {
413 // FIXME need a simpler mechanism for constructing null parameters
414 Q const* q = nullptr;
415 if (!qparams.empty())
416 q = &qparams.front().get();
417 else if (!dparams.empty())
418 q = &dparams.front().get();
419 if (q) {
420 for (size_t i = 0; i < nD; ++i) {
421 dparams_new.emplace_back(handler.copy(*q));
422 dactions_new.emplace_back(handler.copy(*q));
423 handler.fill(0, dparams_new.back());
424 handler.fill(0, dactions_new.back());
425 }
426 }
427 }
428 for (size_t i = 0; i < nD; ++i) {
429 for (size_t j = 0; j < q_delete.size(); ++j) {
430 handler.axpy(solutions_proj(i, j), qparams.at(q_delete[j]), dparams_new.at(i));
431 handler.axpy(solutions_proj(i, j), qactions.at(q_delete[j]), dactions_new.at(i));
432 }
433 for (size_t j = 0; j < dims.nD; ++j) {
434 handler.axpy(solutions_proj(i, nQd + j), dparams.at(j), dparams_new.at(i));
435 handler.axpy(solutions_proj(i, nQd + j), dactions.at(j), dactions_new.at(i));
436 }
437 }
438 for (size_t i = 0; i < nD; ++i) {
439 auto norm = std::sqrt(std::abs(handler.dot(dparams_new.at(i), dparams_new.at(i))));
440 if (norm < norm_thresh) {
441 logger.warn(
442 std::format("construct_dspace: skipping normalisation of D vector {} with near-zero norm = {:.2e}", i, norm));
443 continue;
444 }
445 handler.scal(1. / norm, dparams_new[i]);
446 handler.scal(1. / norm, dactions_new[i]);
447 }
448 return std::make_tuple(std::move(dparams_new), std::move(dactions_new));
449}
450
467template <class R, class Q, class P, typename value_type, typename value_type_abs>
469 const subspace::Dimensions& dims, const CVecRef<P>& pparams, const CVecRef<Q>& qparams,
470 const CVecRef<Q>& dparams, const value_type_abs norm_thresh,
471 ArrayHandlers<R, Q, P>& handlers, Logger& logger) {
472 logger.trace("modified_gram_schmidt()");
473 const auto nR = rparams.size(), nP = pparams.size(), nQ = qparams.size(), nD = dparams.size();
474 // std::cout << "modified_gram_schmidt() nR="<<nR<<" nQ="<<nQ<<" nD="<<nD<<std::endl;
475 assert(nP == dims.nP && nQ == dims.nQ && nD == dims.nD);
476 auto orthogonalise = [&overlap, &rparams, nR](const auto& xparams, auto& handler, const size_t oX, const size_t nX) {
477 for (size_t i = 0; i < nX; ++i) {
478 auto norm = std::abs(overlap(oX + i, oX + i));
479 if (nR > 0) {
480 auto dot_mat = handler.gemm_inner(cwrap(rparams), cwrap_arg(xparams.at(i).get()));
481 std::pair<size_t, size_t> mcoeff_dim = std::make_pair(1, nR);
482 subspace::Matrix<typename std::decay_t<decltype(dot_mat)>::value_type> mcoeff(dot_mat.data(), mcoeff_dim);
483 for (size_t j = 0; j < nR; ++j) {
484 mcoeff(0, j) = -mcoeff(0, j) / norm;
485 }
486 handler.gemm_outer(mcoeff, cwrap_arg(xparams.at(i).get()), rparams);
487 }
488 }
489 };
490 auto prof = molpro::Profiler::single();
491 prof->start("orthoganalise");
492 orthogonalise(pparams, handlers.rp(), dims.oP, nP);
493 orthogonalise(qparams, handlers.rq(), dims.oQ, nQ);
494 orthogonalise(dparams, handlers.rq(), dims.oD, nD);
495 prof->stop();
496 prof->start("get null_params");
497 auto null_params = std::vector<int>{};
498 for (size_t i = 0; i < nR; ++i) {
499 auto norm = std::sqrt(std::abs(handlers.rr().dot(rparams[i], rparams[i])));
500 if (norm > norm_thresh) {
501 handlers.rr().scal(1. / norm, rparams[i]);
502 for (size_t j = i + 1; j < nR; ++j) {
503 auto ov = handlers.rr().dot(rparams[i], rparams[j]);
504 handlers.rr().axpy(-ov, rparams[i], rparams[j]);
505 }
506 } else {
507 null_params.push_back(i);
508 }
509 }
510 prof->stop();
511 return null_params;
512}
513
515template <class R>
516auto get_new_working_set(const std::vector<int>& working_set, const CVecRef<R>& params, const CVecRef<R>& wparams) {
517 auto new_indices = find_ref(wparams, params);
518 auto new_working_set = std::vector<int>{};
519 for (auto i : new_indices) {
520 new_working_set.emplace_back(working_set.at(i));
521 }
522 return new_working_set;
523}
524
553template <class R, class Q, class P>
554auto propose_rspace(IterativeSolver<R, Q, P>& solver, const VecRef<R>& parameters, const VecRef<R>& residuals,
556 ArrayHandlers<R, Q, P>& handlers, Logger& logger,
557 const RSpaceOptions<typename array::ArrayHandler<R, R>::value_type_abs>& r_opts,
558 const QSpaceOptions& q_opts, molpro::profiler::Profiler& profiler) {
559 // auto prof = profiler.push("itsolv::propose_rspace"); // FIXME two separate profilers
560 auto prof = molpro::Profiler::single();
561 logger.trace("itsolv::detail::propose_rspace");
562 logger.trace("dimensions {nP, nQ, nD, nW} = ", xspace.dimensions().nP, xspace.dimensions().nQ, xspace.dimensions().nD,
563 solver.working_set().size());
564 profiler.start("itsolv::ISubspaceSolver::solutions");
565 auto solutions = subspace_solver.solutions();
566 profiler.stop();
567 auto q_delete = limit_qspace_size(xspace.dimensions(), q_opts, solutions, logger);
568 logger.debug("delete Q parameter indices = ", q_delete);
569 if (!q_delete.empty()) {
570 auto prof = profiler.push("construct_dspace");
571 auto [dparams, dactions] =
572 construct_dspace(solutions, xspace, q_delete, r_opts.norm_thresh, r_opts.svd_thresh, handlers.qq(), logger);
573 std::sort(begin(q_delete), end(q_delete), std::greater<int>());
574 for (auto iq : q_delete)
575 xspace.eraseq(iq);
576 auto wdparams = wrap(dparams);
577 auto wdactions = wrap(dactions);
578 xspace.update_dspace(wdparams, wdactions);
579 auto eigenvalues_ref = subspace_solver.eigenvalues();
580 subspace_solver.solve(xspace, solutions.rows());
581 auto eigval_error = std::vector<double>{};
582 std::transform(std::begin(eigenvalues_ref), std::end(eigenvalues_ref), std::begin(subspace_solver.eigenvalues()),
583 std::back_inserter(eigval_error), [](auto& e_ref, auto& e_new) { return std::abs(e_ref - e_new); });
584 logger.debug("eigenvalue error due to new D space = ", eigval_error);
585 // FIXME Optionally, solve the subspace problem again and get an estimate of the error due to new D
586 }
587 // Use modified GS to orthonormalise z against P+Q+D, removing any null parameters.
588 auto wresidual = wrap(residuals.begin(), residuals.begin() + solver.working_set().size());
589 profiler.start("normalise");
590 normalise(wresidual, handlers.rr(), logger);
591 profiler.stop();
592 profiler.start("append_overlap_with_r");
593 prof->start("append_overlap_with_r");
594 const auto full_overlap =
595 append_overlap_with_r(xspace.data.at(subspace::EqnData::S), cwrap(wresidual), xspace.cparamsp(),
596 xspace.cparamsq(), xspace.cparamsd(), handlers, logger);
597 profiler.stop();
598 prof->stop();
599 prof->start("redundant_indices");
600 auto redundant_indices =
601 redundant_parameters(full_overlap, xspace.dimensions().nX, wresidual.size(), r_opts.svd_thresh, logger);
602 prof->stop();
603 logger.debug("redundant indices = ", redundant_indices);
604 util::delete_parameters(redundant_indices, wresidual);
605 profiler.start("modified_gram_schmidt");
606 prof->start("modified_gram_schmidt");
607 auto null_param_indices =
608 modified_gram_schmidt(wresidual, xspace.data.at(subspace::EqnData::S), xspace.dimensions(), xspace.cparamsp(),
609 xspace.cparamsq(), xspace.cparamsd(), r_opts.norm_thresh, handlers, logger);
610 profiler.stop();
611 prof->stop();
612 // Now that there is SVD null_param_indices should always be empty
613 logger.debug("null parameters = ", null_param_indices);
614 util::delete_parameters(null_param_indices, wresidual);
615 normalise(wresidual, handlers.rr(), logger);
616 for (size_t i = 0; i < wresidual.size(); ++i)
617 handlers.rr().copy(parameters.at(i), wresidual.at(i));
618 profiler.start("get_new_working_set");
619 auto new_working_set = get_new_working_set(solver.working_set(), cwrap(residuals), cwrap(wresidual));
620 profiler.stop();
621 return new_working_set;
622}
623} // namespace molpro::linalg::itsolv::detail
624
625#endif // LINEARALGEBRA_SRC_MOLPRO_LINALG_ITSOLV_PROPOSE_RSPACE_H
Enhances various operations between pairs of arrays and allows dynamic code injection with uniform in...
Definition: ArrayHandler.h:162
virtual value_type dot(const AL &x, const AR &y)=0
virtual void scal(value_type alpha, AL &x)=0
virtual AL copy(const AR &source)=0
virtual void axpy(value_type alpha, const AR &x, AL &y)=0
virtual void fill(value_type alpha, AL &x)=0
decltype(check_abs< value_type >()) value_type_abs
Definition: ArrayHandler.h:182
Non-owning container taking a pointer to the data buffer and its size and exposing routines for itera...
Definition: Span.h:31
Class, containing a collection of array handlers used in IterativeSolver Provides a Builder sub-class...
Definition: ArrayHandlers.h:25
auto & qq()
Definition: ArrayHandlers.h:40
auto & rp()
Definition: ArrayHandlers.h:44
auto & rr()
Definition: ArrayHandlers.h:39
auto & rq()
Definition: ArrayHandlers.h:42
Base class defining the interface common to all iterative solvers.
Definition: IterativeSolver.h:209
virtual const std::vector< int > & working_set() const =0
Working set of roots that are not yet converged.
Definition: Logger.h:442
void info(std::string_view message, Ts &&...args) const
Definition: Logger.h:505
void warn(std::string_view message, Ts &&...args) const
Definition: Logger.h:510
void debug(std::string_view message, Ts &&...args) const
Definition: Logger.h:500
void data_dump(std::string_view what, Ts...data) const
Definition: Logger.h:526
void trace(std::string_view message, Ts &&...args) const
Definition: Logger.h:495
SubspaceData< value_type > data
Equation data in the subspace.
Definition: IXSpace.h:28
virtual CVecRef< Q > cparamsq() const =0
virtual CVecRef< P > cparamsp() const =0
virtual CVecRef< Q > cactionsd() const =0
virtual void eraseq(size_t i)=0
Removes parameter i from Q subspace.
virtual CVecRef< Q > cactionsq() const =0
virtual CVecRef< Q > cparamsd() const =0
virtual const Dimensions & dimensions() const =0
virtual void update_dspace(VecRef< Q > &params, VecRef< Q > &actions)=0
Updates D space with the new parameters.
Slice row(size_t i)
Access row slice.
Definition: Matrix.h:120
void remove_row(index_type row)
removes a row from the matrix
Definition: Matrix.h:144
const std::vector< T > & data() const &
Access the underlying data buffer.
Definition: Matrix.h:67
size_t size() const
Definition: Matrix.h:170
index_type rows() const
Definition: Matrix.h:167
index_type cols() const
Definition: Matrix.h:168
Profiler & start(const std::string &name)
Profiler & stop(const std::string &name="")
Proxy push(const std::string &name)
static std::shared_ptr< Profiler > single()
auto construct_projected_solution(const subspace::Matrix< value_type > &solutions, const subspace::Dimensions &dims, const std::vector< std::size_t > &remove_qspace, Logger &logger)
Projects solution from the full subspace on to Q_{delete} and current D space.
Definition: propose_rspace.h:53
auto construct_projected_solutions_overlap(const subspace::Matrix< value_type > &solutions_proj, const subspace::Matrix< value_type > &overlap, const subspace::Dimensions &dims, const std::vector< std::size_t > &remove_qspace, Logger &logger)
Constructs overlap matrix for projected solutions.
Definition: propose_rspace.h:86
auto remove_null_projected_solutions(const subspace::Matrix< value_type > &solutions_proj, const subspace::Matrix< value_type > &overlap_proj, const value_type_abs svd_thresh, Logger &logger)
Transforms to a stable subspace of projected solutions via SVD.
Definition: propose_rspace.h:169
void remove_null_norm_and_normalise(subspace::Matrix< value_type > &parameters, subspace::Matrix< value_type > &overlap, const value_type_abs norm_thresh, Logger &logger)
Removes parameters with norm less than threshold and normalises the rest.
Definition: propose_rspace.h:130
auto construct_full_subspace_overlap(const subspace::Matrix< value_type > &solutions_proj, const subspace::Dimensions &dims, const std::vector< std::size_t > &remove_qspace, const subspace::Matrix< value_type > &overlap, const size_t nR)
Constructs overlap matrix of P+Q+R+(projected solutions) subspaces, where Q is without removed parame...
Definition: propose_rspace.h:201
Definition: helper-dispatch.h:140
void normalise(const size_t n_roots, const VecRef< R > &params, const VecRef< R > &actions, array::ArrayHandler< R, R > &handler, Logger &logger)
Definition: IterativeSolverTemplate.h:84
auto propose_rspace(IterativeSolver< R, Q, P > &solver, const VecRef< R > &parameters, const VecRef< R > &residuals, subspace::IXSpace< R, Q, P > &xspace, subspace::ISubspaceSolver< R, Q, P > &subspace_solver, ArrayHandlers< R, Q, P > &handlers, Logger &logger, const RSpaceOptions< typename array::ArrayHandler< R, R >::value_type_abs > &r_opts, const QSpaceOptions &q_opts, molpro::profiler::Profiler &profiler)
Proposes new parameters for the subspace from the preconditioned residuals.
Definition: propose_rspace.h:554
auto construct_dspace(const subspace::Matrix< value_type > &solutions, const subspace::IXSpace< R, Q, P > &xspace, const std::vector< std::size_t > &q_delete, const value_type_abs norm_thresh, const value_type_abs svd_thresh, array::ArrayHandler< Q, Q > &handler, Logger &logger)
Constructs the new D space by projecting the solutions onto Qd+D subspace and ensuring they are well ...
Definition: propose_rspace.h:391
auto get_new_working_set(const std::vector< int > &working_set, const CVecRef< R > &params, const CVecRef< R > &wparams)
Returns new working set based on parameters included in wparams.
Definition: propose_rspace.h:516
auto append_overlap_with_r(const subspace::Matrix< value_type > &overlap, const CVecRef< R > &params, const CVecRef< P > &pparams, const CVecRef< Q > &qparams, const CVecRef< Q > &dparams, ArrayHandlers< R, Q, P > &handlers, Logger &logger)
Constructs overlap of the full subspace by appending overlap with new parameters to the overlap of pr...
Definition: propose_rspace.h:282
std::vector< std::size_t > limit_qspace_size(const subspace::Dimensions &dims, const QSpaceOptions &opts, const subspace::Matrix< value_type > &solutions, Logger &logger)
Ensures that size Q space is within limit by proposing Q parameters for deletion.
Definition: propose_rspace.h:319
auto modified_gram_schmidt(const VecRef< R > &rparams, const subspace::Matrix< value_type > &overlap, const subspace::Dimensions &dims, const CVecRef< P > &pparams, const CVecRef< Q > &qparams, const CVecRef< Q > &dparams, const value_type_abs norm_thresh, ArrayHandlers< R, Q, P > &handlers, Logger &logger)
Orthogonalises R parameters against P+Q+D subspace (and themselves)
Definition: propose_rspace.h:468
auto redundant_parameters(const subspace::Matrix< value_type > &overlap, const size_t oR, const size_t nR, const value_type_abs svd_thresh, Logger &logger)
Deduces a set of parameters that are redundant due to linear dependencies.
Definition: helper-implementation.h:627
auto overlap(const CVecRef< R > &left, const CVecRef< Q > &right, array::ArrayHandler< Z, W > &handler) -> std::enable_if_t< detail::Z_and_W_are_one_of_R_and_Q< R, Q, Z, W >, Matrix< typename array::ArrayHandler< Z, W >::value_type > >
Calculates overlap matrix between left and right vectors.
Definition: util.h:51
void delete_parameters(std::vector< int > indices, Container &params)
Removes parameters.
Definition: util.h:98
auto cwrap(ForwardIt begin, ForwardIt end)
Takes a begin and end iterators and returns a vector of references to each element.
Definition: wrap.h:52
auto cwrap_arg(T &&arg, S &&... args) -> std::enable_if_t< std::conjunction_v< std::is_same< decay_t< T >, decay_t< S > >... >, CVecRef< decay_t< T > > >
Constructs a vector of const reference wrappers with provided arguments.
Definition: wrap.h:149
auto wrap(ForwardIt begin, ForwardIt end)
Takes a begin and end iterators and returns a vector of references to each element.
Definition: wrap.h:32
std::vector< std::reference_wrapper< const A > > CVecRef
Definition: wrap.h:14
std::list< SVD< value_type > > svd_system(size_t nrows, size_t ncols, const array::Span< value_type > &m, real_type_t< value_type > threshold, bool hermitian=false, bool reduce_to_rank=false)
Performs singular value decomposition and returns SVD objects for singular values less than threshold...
Definition: helper-implementation.h:213
std::vector< std::reference_wrapper< A > > VecRef
Definition: wrap.h:11
std::vector< size_t > find_ref(const VecRef< R > &wparams, ForwardIt begin, ForwardIt end)
Given wrapped references in wparams and a range of original parameters [begin, end),...
Definition: wrap_util.h:17
real_type_t< value_type > precision_scaled(double tolerance_for_double)
Rescale a tolerance that was calibrated for IEEE double precision to the working precision.
Definition: scalar_traits.h:54
Stores partitioning of XSpace into P, Q and R blocks with sizes and offsets for each one.
Definition: Dimensions.h:8
size_t oQ
Definition: Dimensions.h:16
size_t nD
Definition: Dimensions.h:13
size_t nX
Definition: Dimensions.h:14
size_t nQ
Definition: Dimensions.h:12
size_t nP
Definition: Dimensions.h:11
size_t oP
Definition: Dimensions.h:15
size_t oD
Definition: Dimensions.h:17
Manages solution of the subspace problem and storage of those solutions.
Definition: ISubspaceSolver.h:21
virtual void solve(IXSpace< R, Q, P > &xspace, size_t nroots_max)=0
Solve the subspace problem.
virtual const std::vector< value_type > & eigenvalues() const =0
Access eigenvalues from the last solve() call.
virtual const Matrix< value_type > & solutions() const =0
Access solutions from the last solve() call.