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, double thresh = 1.0e-14) {
30 for (auto& p : params) {
31 auto dot = handler.dot(p, p);
32 dot = std::sqrt(std::abs(dot));
33 if (dot > thresh) {
34 handler.scal(1. / dot, p);
35 } else {
36 logger.warn("parameter's length is too small for normalisation, dot = " + std::format("{:.2e}", dot));
37 }
38 }
39}
40namespace dspace {
41
50template <typename value_type>
52 const std::vector<std::size_t>& remove_qspace, Logger& logger) {
53 logger.trace("construct_projected_solution()");
54 const auto nQd = remove_qspace.size();
55 const auto nSol = solutions.rows();
56 auto solutions_proj = subspace::Matrix<value_type>({nSol, nQd + dims.nD});
57 for (size_t i = 0; i < nSol; ++i) {
58 for (size_t j = 0; j < nQd; ++j) {
59 solutions_proj(i, j) = solutions(i, dims.oQ + remove_qspace[j]);
60 }
61 for (size_t j = 0; j < dims.nD; ++j) {
62 solutions_proj(i, nQd + j) = solutions(i, dims.oD + j);
63 }
64 }
65 logger.debug("nSol, nQd, nD", nSol, nQd, dims.nD);
66 return solutions_proj;
67}
68
83template <typename value_type>
85 const subspace::Matrix<value_type>& overlap,
86 const subspace::Dimensions& dims,
87 const std::vector<std::size_t>& remove_qspace, Logger& logger) {
88 logger.trace("construct_projected_solution_overlap()");
89 const auto nSol = solutions_proj.rows();
90 const auto nQd = remove_qspace.size();
91 auto overlap_proj = subspace::Matrix<value_type>({nSol, nSol});
92 for (size_t i = 0; i < nSol; ++i) {
93 for (size_t ii = 0; ii <= i; ++ii) {
94 for (size_t j = 0; j < nQd; ++j) {
95 for (size_t k = 0; k < nQd; ++k) {
96 overlap_proj(i, ii) += solutions_proj(i, j) * solutions_proj(ii, k) *
97 overlap(dims.oQ + remove_qspace[j], dims.oQ + remove_qspace[k]);
98 }
99 for (size_t k = 0; k < dims.nD; ++k) {
100 overlap_proj(i, ii) +=
101 solutions_proj(i, j) * solutions_proj(ii, nQd + k) * overlap(dims.oQ + remove_qspace[j], dims.oD + k);
102 }
103 }
104 for (size_t j = 0; j < dims.nD; ++j) {
105 for (size_t k = 0; k < dims.nD; ++k) {
106 overlap_proj(i, ii) +=
107 solutions_proj(i, nQd + j) * solutions_proj(ii, nQd + k) * overlap(dims.oD + j, dims.oD + k);
108 }
109 for (size_t k = 0; k < nQd; ++k) {
110 overlap_proj(i, ii) +=
111 solutions_proj(i, nQd + j) * solutions_proj(ii, k) * overlap(dims.oD + j, dims.oQ + remove_qspace[k]);
112 }
113 }
114 overlap_proj(ii, i) = overlap_proj(i, ii);
115 }
116 }
117 return overlap_proj;
118}
119
127template <typename value_type, typename value_type_abs>
129 const value_type_abs norm_thresh, Logger& logger) {
130 logger.trace("remove_null_norm_and_normalise()");
131 const auto nSol = parameters.rows();
132 auto norm_proj = std::vector<value_type_abs>(nSol, 0.);
133 for (size_t i = 0; i < nSol; ++i)
134 norm_proj[i] = std::sqrt(std::abs(overlap(i, i)));
135 for (size_t i = 0, j = 0; i < nSol; ++i) {
136 if (norm_proj[i] > norm_thresh) {
137 parameters.row(j).scal(1. / norm_proj[i]);
138 overlap.col(j).scal(1. / norm_proj[i]);
139 overlap.row(j).scal(1. / norm_proj[i]);
140 ++j;
141 } else {
142 parameters.remove_row(j);
143 overlap.remove_row_col(j, j);
144 std::stringstream ss;
145 ss << std::setprecision(3) << "remove projected solution parameter i = " << i << ", norm = " << norm_proj[i];
146 logger.info(ss.str());
147 }
148 }
149 logger.data_dump("norm = ", norm_proj);
150 logger.data_dump("parameters after normalisation = ", parameters);
151 logger.data_dump("overlap of parameters = ", overlap);
152}
153
166template <typename value_type, typename value_type_abs>
168 const subspace::Matrix<value_type>& overlap_proj, const value_type_abs svd_thresh,
169 Logger& logger) {
170 logger.trace("remove_null_projected_solutions()");
171 logger.debug("nS on entry = ", solutions_proj.rows());
172 value_type* m = const_cast<std::vector<value_type>&>(overlap_proj.data()).data();
173 auto svd_vecs = svd_system(overlap_proj.rows(), overlap_proj.cols(), array::Span(m, overlap_proj.size()),
174 std::numeric_limits<value_type_abs>::max(), true);
175 svd_vecs.remove_if([&svd_thresh](const auto& el) { return el.value < svd_thresh; });
176 svd_vecs.sort([](const auto& lt, const auto& rt) { return lt.value < rt.value; });
177 const auto nD = svd_vecs.size();
178 const auto nX = solutions_proj.cols();
179 auto solutions_stable = subspace::Matrix<value_type>({nD, nX});
180 auto svd = svd_vecs.begin();
181 for (size_t i = 0; i < nD; ++i, ++svd)
182 for (size_t j = 0; j < overlap_proj.cols(); ++j)
183 for (size_t k = 0; k < nX; ++k)
184 solutions_stable(i, k) += svd->v[j] * solutions_proj(j, k);
185 logger.debug("nS without null space = ", solutions_stable.rows());
186 return solutions_stable;
187}
188
198template <typename value_type>
200 const subspace::Dimensions& dims, const std::vector<std::size_t>& remove_qspace,
201 const subspace::Matrix<value_type>& overlap, const size_t nR) {
202 const auto nDnew = solutions_proj.rows();
203 const auto nQd = remove_qspace.size();
204 const auto nQ = dims.nQ - nQd;
205 auto ov = overlap;
206 for (size_t i = 0; i < dims.nD; ++i) {
207 ov.remove_row_col(dims.oD, dims.oD);
208 }
209 auto is_Qdelete = [&remove_qspace](size_t i) {
210 return std::find(begin(remove_qspace), end(remove_qspace), i) != end(remove_qspace);
211 };
212 for (size_t i = 0, j = 0; i < dims.nQ; ++i) {
213 if (is_Qdelete(i))
214 ov.remove_row_col(dims.oQ + j, dims.oQ + j);
215 else
216 ++j;
217 }
218 const auto oDnew = dims.nP + nQ + nR;
219 ov.resize({oDnew + nDnew, oDnew + nDnew});
220 /*
221 * append overlap with solutions_proj
222 * x_i = \sum_j c_ij v_j
223 * <x_i, v_j> = \sum_k c_ik <v_k, v_j>
224 * <x_i, x_j> = \sum_kl c_ik c_jl <v_k, v_l>
225 */
226 auto accumulate_ov_offdiag = [&](size_t i, size_t j, size_t jj) {
227 for (size_t k = 0; k < nQd; ++k)
228 ov(oDnew + i, j) += solutions_proj(i, k) * overlap(jj, dims.oQ + remove_qspace[k]);
229 for (size_t k = 0; k < dims.nD; ++k)
230 ov(oDnew + i, j) += solutions_proj(i, nQd + k) * overlap(jj, dims.oD + k);
231 ov(j, oDnew + i) = ov(oDnew + i, j);
232 };
233 for (size_t i = 0; i < nDnew; ++i) {
234 for (size_t j = 0; j < dims.nP; ++j)
235 accumulate_ov_offdiag(i, j, dims.oP + j);
236 for (size_t j = 0, jj = 0; j < dims.nQ; ++j)
237 if (!is_Qdelete(j))
238 accumulate_ov_offdiag(i, dims.nP + jj++, dims.oQ + j);
239 for (size_t j = 0; j < nR; ++j)
240 accumulate_ov_offdiag(i, dims.nP + nQ + j, dims.nX + j);
241 }
242 for (size_t i = 0; i < nDnew; ++i) {
243 for (size_t j = 0; j <= i; ++j) {
244 for (size_t k = 0; k < nQd; ++k) {
245 for (size_t l = 0; l < nQd; ++l)
246 ov(oDnew + i, oDnew + j) += solutions_proj(i, k) * solutions_proj(j, l) *
247 overlap(dims.oQ + remove_qspace[k], dims.oQ + remove_qspace[l]);
248 for (size_t l = 0; l < dims.nD; ++l)
249 ov(oDnew + i, oDnew + j) +=
250 solutions_proj(i, k) * solutions_proj(j, nQd + l) * overlap(dims.oQ + remove_qspace[k], dims.oD + l);
251 }
252 for (size_t k = 0; k < dims.nD; ++k) {
253 for (size_t l = 0; l < nQd; ++l)
254 ov(oDnew + i, oDnew + j) +=
255 solutions_proj(i, nQd + k) * solutions_proj(j, l) * overlap(dims.oD + k, dims.oQ + remove_qspace[l]);
256 for (size_t l = 0; l < dims.nD; ++l)
257 ov(oDnew + i, oDnew + j) +=
258 solutions_proj(i, nQd + k) * solutions_proj(j, nQd + l) * overlap(dims.oD + k, dims.oD + l);
259 }
260 ov(oDnew + j, oDnew + i) = ov(oDnew + i, oDnew + j);
261 }
262 }
263 return ov;
264}
265} // namespace dspace
266
279template <class R, class Q, class P, typename value_type>
281 const CVecRef<P>& pparams, const CVecRef<Q>& qparams, const CVecRef<Q>& dparams,
282 ArrayHandlers<R, Q, P>& handlers, Logger& logger) {
283 logger.trace("append_overlap_with_r()");
284 const auto nP = pparams.size(), nQ = qparams.size(), nD = dparams.size(), nN = params.size();
285 const auto nX = nP + nQ + nD + nN;
286 const auto oP = 0;
287 const auto oQ = oP + nP;
288 const auto oD = oQ + nQ;
289 const auto oN = oD + nD;
290 auto ov = overlap;
291 ov.resize({nX, nX}); // solutions are last
292 ov.slice({oN, oN}, {oN + nN, oN + nN}) = subspace::util::overlap(params, handlers.rr());
293 ov.slice({oN, oP}, {oN + nN, oP + nP}) = subspace::util::overlap(params, pparams, handlers.rp());
294 ov.slice({oN, oQ}, {oN + nN, oQ + nQ}) = subspace::util::overlap(params, qparams, handlers.rq());
295 ov.slice({oN, oD}, {oN + nN, oD + nD}) = subspace::util::overlap(params, dparams, handlers.rq());
296 auto copy_upper_to_lower = [&ov, oN, nN](size_t oX, size_t nX) {
297 for (size_t i = 0; i < nX; ++i)
298 for (size_t j = 0; j < nN; ++j)
299 ov(oX + i, oN + j) = ov(oN + j, oX + i);
300 };
301 copy_upper_to_lower(oP, nP);
302 copy_upper_to_lower(oQ, nQ);
303 copy_upper_to_lower(oD, nD);
304 logger.data_dump("full overlap P+Q+D+R = ", ov);
305 return ov;
306}
307
316template <typename value_type>
317std::vector<std::size_t> limit_qspace_size(const subspace::Dimensions& dims, const QSpaceOptions& opts,
318 const subspace::Matrix<value_type>& solutions, Logger& logger) {
319 using value_type_abs = decltype(std::abs(solutions(0, 0)));
320 using contrib_pair = std::pair<std::size_t, value_type_abs>;
321 using std::begin;
322 using std::end;
323
324 logger.trace("limit_qspace_size()");
325
326 assert(opts.max_size >= opts.min_size);
327
328 const bool must_trim_size = dims.nQ > opts.max_size;
329 const bool may_trim_size = dims.nQ > opts.min_size && opts.contrib_thresh > 0;
330
331 if (!must_trim_size && !may_trim_size) {
332 return {};
333 }
334
335 const std::size_t min_deletions = must_trim_size ? dims.nQ - opts.max_size : 0;
336 const std::size_t max_deletions = dims.nQ - opts.min_size;
337
338 std::vector<std::size_t> q_delete;
339 q_delete.reserve(min_deletions);
340
341 const std::size_t nSol = solutions.rows();
342
343 std::vector<contrib_pair> max_contrib_to_solution;
344 max_contrib_to_solution.reserve(nSol);
345
346 for (std::size_t idx : std::ranges::views::iota(std::size_t(0), dims.nQ)) {
347 max_contrib_to_solution.emplace_back(std::make_pair(idx, value_type_abs(0)));
348
349 for (size_t j = 0; j < nSol; ++j) {
350 const value_type_abs current = std::abs(solutions(j, dims.oQ + idx));
351 max_contrib_to_solution.back().second = std::max(max_contrib_to_solution.back().second, current);
352 }
353 }
354
355 std::ranges::sort(max_contrib_to_solution, std::less<>{}, &contrib_pair::second);
356
357 logger.trace("contribution to solutions =",
358 max_contrib_to_solution | std::ranges::views::transform([](const contrib_pair& p) { return p.second; }));
359
360 for (auto [idx, contrib] : max_contrib_to_solution) {
361 if (q_delete.size() == max_deletions || (q_delete.size() >= min_deletions && contrib >= opts.contrib_thresh)) {
362 break;
363 }
364
365 if (opts.contrib_thresh != 0 && contrib >= opts.contrib_thresh) {
366 logger.warn("deleted Q vector with max. contribution = ", contrib);
367 } else {
368 logger.debug("deleted Q vector with max. contribution = ", contrib);
369 }
370
371 q_delete.emplace_back(idx);
372 }
373
374 return q_delete;
375}
376
388template <class R, class Q, class P, typename value_type, typename value_type_abs>
390 const std::vector<std::size_t>& q_delete, const value_type_abs norm_thresh,
391 const value_type_abs svd_thresh, array::ArrayHandler<Q, Q>& handler, Logger& logger) {
392 const auto dims = xspace.dimensions();
393 const auto overlap = xspace.data.at(subspace::EqnData::S);
394 auto solutions_proj = dspace::construct_projected_solution(solutions, dims, q_delete, logger);
395 auto overlap_proj = dspace::construct_projected_solutions_overlap(solutions_proj, overlap, dims, q_delete, logger);
396 dspace::remove_null_norm_and_normalise(solutions_proj, overlap_proj, norm_thresh, logger);
397 solutions_proj = dspace::remove_null_projected_solutions(solutions_proj, overlap_proj, svd_thresh, logger);
398 overlap_proj = dspace::construct_projected_solutions_overlap(solutions_proj, overlap, dims, q_delete, logger);
399 dspace::remove_null_norm_and_normalise(solutions_proj, overlap_proj, norm_thresh, logger);
400 auto overlap_full_subspace = dspace::construct_full_subspace_overlap(solutions_proj, dims, q_delete, overlap, 0);
401 auto svd_vecs = svd_system(overlap_full_subspace.rows(), overlap_full_subspace.cols(),
402 array::Span(&overlap_full_subspace(0, 0), overlap_full_subspace.size()), svd_thresh, true);
403 assert(svd_vecs.empty() && "P+Q+D subspace should be stable by construction");
404 const auto nD = solutions_proj.rows();
405 const auto nQd = q_delete.size();
406 assert(nQd + dims.nD == solutions_proj.cols());
407 const auto &qparams = xspace.cparamsq(), qactions = xspace.cactionsq();
408 const auto &dparams = xspace.cparamsd(), dactions = xspace.cactionsd();
409 std::vector<Q> dparams_new, dactions_new;
410 {
411 // FIXME need a simpler mechanism for constructing null parameters
412 Q const* q = nullptr;
413 if (!qparams.empty())
414 q = &qparams.front().get();
415 else if (!dparams.empty())
416 q = &dparams.front().get();
417 if (q) {
418 for (size_t i = 0; i < nD; ++i) {
419 dparams_new.emplace_back(handler.copy(*q));
420 dactions_new.emplace_back(handler.copy(*q));
421 handler.fill(0, dparams_new.back());
422 handler.fill(0, dactions_new.back());
423 }
424 }
425 }
426 for (size_t i = 0; i < nD; ++i) {
427 for (size_t j = 0; j < q_delete.size(); ++j) {
428 handler.axpy(solutions_proj(i, j), qparams.at(q_delete[j]), dparams_new.at(i));
429 handler.axpy(solutions_proj(i, j), qactions.at(q_delete[j]), dactions_new.at(i));
430 }
431 for (size_t j = 0; j < dims.nD; ++j) {
432 handler.axpy(solutions_proj(i, nQd + j), dparams.at(j), dparams_new.at(i));
433 handler.axpy(solutions_proj(i, nQd + j), dactions.at(j), dactions_new.at(i));
434 }
435 }
436 for (size_t i = 0; i < nD; ++i) {
437 auto norm = std::sqrt(std::abs(handler.dot(dparams_new.at(i), dparams_new.at(i))));
438 if (norm < norm_thresh) {
439 logger.warn(
440 std::format("construct_dspace: skipping normalisation of D vector {} with near-zero norm = {:.2e}", i, norm));
441 continue;
442 }
443 handler.scal(1. / norm, dparams_new[i]);
444 handler.scal(1. / norm, dactions_new[i]);
445 }
446 return std::make_tuple(std::move(dparams_new), std::move(dactions_new));
447}
448
465template <class R, class Q, class P, typename value_type, typename value_type_abs>
467 const subspace::Dimensions& dims, const CVecRef<P>& pparams, const CVecRef<Q>& qparams,
468 const CVecRef<Q>& dparams, const value_type_abs norm_thresh,
469 ArrayHandlers<R, Q, P>& handlers, Logger& logger) {
470 logger.trace("modified_gram_schmidt()");
471 const auto nR = rparams.size(), nP = pparams.size(), nQ = qparams.size(), nD = dparams.size();
472 // std::cout << "modified_gram_schmidt() nR="<<nR<<" nQ="<<nQ<<" nD="<<nD<<std::endl;
473 assert(nP == dims.nP && nQ == dims.nQ && nD == dims.nD);
474 auto orthogonalise = [&overlap, &rparams, nR](const auto& xparams, auto& handler, const size_t oX, const size_t nX) {
475 for (size_t i = 0; i < nX; ++i) {
476 auto norm = std::abs(overlap(oX + i, oX + i));
477 if (nR > 0) {
478 auto dot_mat = handler.gemm_inner(cwrap(rparams), cwrap_arg(xparams.at(i).get()));
479 std::pair<size_t, size_t> mcoeff_dim = std::make_pair(1, nR);
480 subspace::Matrix<double> mcoeff(dot_mat.data(), mcoeff_dim);
481 for (size_t j = 0; j < nR; ++j) {
482 mcoeff(0, j) = -mcoeff(0, j) / norm;
483 }
484 handler.gemm_outer(mcoeff, cwrap_arg(xparams.at(i).get()), rparams);
485 }
486 }
487 };
488 auto prof = molpro::Profiler::single();
489 prof->start("orthoganalise");
490 orthogonalise(pparams, handlers.rp(), dims.oP, nP);
491 orthogonalise(qparams, handlers.rq(), dims.oQ, nQ);
492 orthogonalise(dparams, handlers.rq(), dims.oD, nD);
493 prof->stop();
494 prof->start("get null_params");
495 auto null_params = std::vector<int>{};
496 for (size_t i = 0; i < nR; ++i) {
497 auto norm = std::sqrt(std::abs(handlers.rr().dot(rparams[i], rparams[i])));
498 if (norm > norm_thresh) {
499 handlers.rr().scal(1. / norm, rparams[i]);
500 for (size_t j = i + 1; j < nR; ++j) {
501 auto ov = handlers.rr().dot(rparams[i], rparams[j]);
502 handlers.rr().axpy(-ov, rparams[i], rparams[j]);
503 }
504 } else {
505 null_params.push_back(i);
506 }
507 }
508 prof->stop();
509 return null_params;
510}
511
513template <class R>
514auto get_new_working_set(const std::vector<int>& working_set, const CVecRef<R>& params, const CVecRef<R>& wparams) {
515 auto new_indices = find_ref(wparams, params);
516 auto new_working_set = std::vector<int>{};
517 for (auto i : new_indices) {
518 new_working_set.emplace_back(working_set.at(i));
519 }
520 return new_working_set;
521}
522
551template <class R, class Q, class P>
552auto propose_rspace(IterativeSolver<R, Q, P>& solver, const VecRef<R>& parameters, const VecRef<R>& residuals,
554 ArrayHandlers<R, Q, P>& handlers, Logger& logger, const RSpaceOptions& r_opts,
555 const QSpaceOptions& q_opts, molpro::profiler::Profiler& profiler) {
556 // auto prof = profiler.push("itsolv::propose_rspace"); // FIXME two separate profilers
557 auto prof = molpro::Profiler::single();
558 logger.trace("itsolv::detail::propose_rspace");
559 logger.trace("dimensions {nP, nQ, nD, nW} = ", xspace.dimensions().nP, xspace.dimensions().nQ, xspace.dimensions().nD,
560 solver.working_set().size());
561 profiler.start("itsolv::ISubspaceSolver::solutions");
562 auto solutions = subspace_solver.solutions();
563 profiler.stop();
564 auto q_delete = limit_qspace_size(xspace.dimensions(), q_opts, solutions, logger);
565 logger.debug("delete Q parameter indices = ", q_delete);
566 if (!q_delete.empty()) {
567 auto prof = profiler.push("construct_dspace");
568 auto [dparams, dactions] =
569 construct_dspace(solutions, xspace, q_delete, r_opts.norm_thresh, r_opts.svd_thresh, handlers.qq(), logger);
570 std::sort(begin(q_delete), end(q_delete), std::greater<int>());
571 for (auto iq : q_delete)
572 xspace.eraseq(iq);
573 auto wdparams = wrap(dparams);
574 auto wdactions = wrap(dactions);
575 xspace.update_dspace(wdparams, wdactions);
576 auto eigenvalues_ref = subspace_solver.eigenvalues();
577 subspace_solver.solve(xspace, solutions.rows());
578 auto eigval_error = std::vector<double>{};
579 std::transform(std::begin(eigenvalues_ref), std::end(eigenvalues_ref), std::begin(subspace_solver.eigenvalues()),
580 std::back_inserter(eigval_error), [](auto& e_ref, auto& e_new) { return std::abs(e_ref - e_new); });
581 logger.debug("eigenvalue error due to new D space = ", eigval_error);
582 // FIXME Optionally, solve the subspace problem again and get an estimate of the error due to new D
583 }
584 // Use modified GS to orthonormalise z against P+Q+D, removing any null parameters.
585 auto wresidual = wrap(residuals.begin(), residuals.begin() + solver.working_set().size());
586 profiler.start("normalise");
587 normalise(wresidual, handlers.rr(), logger);
588 profiler.stop();
589 profiler.start("append_overlap_with_r");
590 prof->start("append_overlap_with_r");
591 const auto full_overlap =
592 append_overlap_with_r(xspace.data.at(subspace::EqnData::S), cwrap(wresidual), xspace.cparamsp(),
593 xspace.cparamsq(), xspace.cparamsd(), handlers, logger);
594 profiler.stop();
595 prof->stop();
596 prof->start("redundant_indices");
597 auto redundant_indices =
598 redundant_parameters(full_overlap, xspace.dimensions().nX, wresidual.size(), r_opts.svd_thresh, logger);
599 prof->stop();
600 logger.debug("redundant indices = ", redundant_indices);
601 util::delete_parameters(redundant_indices, wresidual);
602 profiler.start("modified_gram_schmidt");
603 prof->start("modified_gram_schmidt");
604 auto null_param_indices =
605 modified_gram_schmidt(wresidual, xspace.data.at(subspace::EqnData::S), xspace.dimensions(), xspace.cparamsp(),
606 xspace.cparamsq(), xspace.cparamsd(), r_opts.norm_thresh, handlers, logger);
607 profiler.stop();
608 prof->stop();
609 // Now that there is SVD null_param_indices should always be empty
610 logger.debug("null parameters = ", null_param_indices);
611 util::delete_parameters(null_param_indices, wresidual);
612 normalise(wresidual, handlers.rr(), logger);
613 for (size_t i = 0; i < wresidual.size(); ++i)
614 handlers.rr().copy(parameters.at(i), wresidual.at(i));
615 profiler.start("get_new_working_set");
616 auto new_working_set = get_new_working_set(solver.working_set(), cwrap(residuals), cwrap(wresidual));
617 profiler.stop();
618 return new_working_set;
619}
620} // namespace molpro::linalg::itsolv::detail
621
622#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
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:202
virtual const std::vector< int > & working_set() const =0
Working set of roots that are not yet converged.
Definition: Logger.h:428
void info(std::string_view message, Ts &&...args) const
Definition: Logger.h:491
void warn(std::string_view message, Ts &&...args) const
Definition: Logger.h:496
void debug(std::string_view message, Ts &&...args) const
Definition: Logger.h:486
void data_dump(std::string_view what, Ts...data) const
Definition: Logger.h:512
void trace(std::string_view message, Ts &&...args) const
Definition: Logger.h:481
virtual CVecRef< Q > cparamsq() const =0
virtual CVecRef< P > cparamsp() const =0
virtual CVecRef< Q > cactionsd() const =0
SubspaceData data
Equation data in the subspace.
Definition: IXSpace.h:28
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:51
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:84
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:167
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:128
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:199
Definition: helper-implementation.h:613
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 &r_opts, const QSpaceOptions &q_opts, molpro::profiler::Profiler &profiler)
Proposes new parameters for the subspace from the preconditioned residuals.
Definition: propose_rspace.h:552
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 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:389
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:514
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:280
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:317
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:466
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:629
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< double > >
Calculates overlap matrix between left and right vectors.
Definition: util.h:49
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::list< SVD< value_type > > svd_system(size_t nrows, size_t ncols, const array::Span< value_type > &m, double 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:268
std::vector< std::reference_wrapper< const A > > CVecRef
Definition: wrap.h:14
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
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.