uni20/ops_2tridiagonal__eigen_8hpp_source.html

#pragma once


#include <uni20/linalg/backends/lapack/tridiagonal_eigen.hpp>

#include <uni20/linalg/dispatch.hpp>

#include <uni20/linalg/operation_tags.hpp>

#include <uni20/tensor/concepts.hpp>


#include <span>

#include <utility>


namespace uni20::linalg

{


template <class BackendSelector, uni20::LapackReal Scalar, uni20::MutableRankedTensorView<2> EigenvectorTensor>


void symmetric_tridiagonal_eigen(BackendSelector&& selector, std::span<Scalar> diagonal, std::span<Scalar> subdiagonal,

                                 EigenvectorTensor&& eigenvectors, bool compute_vectors)

{

  auto eigenvector_descriptor = uni20::mdspec_of(eigenvectors);

  dispatch_kernel(std::forward<BackendSelector>(selector),

                  symmetric_tridiagonal_eigen_op{.compute_vectors = compute_vectors}, diagonal, subdiagonal,

                  eigenvector_descriptor);

}


template <uni20::LapackReal Scalar, uni20::MutableRankedTensorView<2> EigenvectorTensor>


void symmetric_tridiagonal_eigen(std::span<Scalar> diagonal, std::span<Scalar> subdiagonal,

                                 EigenvectorTensor&& eigenvectors, bool compute_vectors)

{

  auto selector = select_backend(symmetric_tridiagonal_eigen_op{.compute_vectors = compute_vectors}, eigenvectors);

  symmetric_tridiagonal_eigen(selector, diagonal, subdiagonal, std::forward<EigenvectorTensor>(eigenvectors),

                              compute_vectors);

}


} // namespace uni20::linalg

tridiagonal_eigen.hpp
LAPACK backend for real symmetric tridiagonal eigensystems.

dispatch.hpp
Operation-tag backend dispatch over tensor-view operands.

uni20::linalg
Definition linalg/async.hpp:28

uni20::linalg::select_backend
constexpr auto select_backend(Operation const &operation, FirstTensor const &, RestTensors const &...)
Select the default backend list for an operation on Tensor operands.
Definition backend_selector.hpp:252

uni20::linalg::dispatch_kernel
void dispatch_kernel(BackendSelector &&selector, Op op, Args &&... args)
Normalize a selector and dispatch or report that every eligible backend declined.
Definition dispatch.hpp:374

uni20::linalg::symmetric_tridiagonal_eigen
void symmetric_tridiagonal_eigen(BackendSelector &&selector, std::span< Scalar > diagonal, std::span< Scalar > subdiagonal, EigenvectorTensor &&eigenvectors, bool compute_vectors)
Overwrite a real symmetric tridiagonal matrix with its eigensystem.
Definition ops/tridiagonal_eigen.hpp:34

uni20::mdspec_of
constexpr decltype(auto) mdspec_of(Tensor &tensor) noexcept(noexcept(tensor.mdspec()))
Return a tensor's immediate or descriptor-backed multidimensional metadata.
Definition tensor/concepts.hpp:50

operation_tags.hpp
Backend-independent operation values used with normalized kernel operands.

uni20::linalg::symmetric_tridiagonal_eigen_op
Symmetric tridiagonal eigensystem operation tag.
Definition operation_tags.hpp:301

uni20::linalg::symmetric_tridiagonal_eigen_op::compute_vectors
bool compute_vectors
Definition operation_tags.hpp:303

concepts.hpp
Concepts for tensor-level objects that resolve mdspan kernel operands.