ops/nonsymmetric_eigen.hpp

Go to the documentation of this file.

#pragma once
 
#include <uni20/linalg/backends/lapack/nonsymmetric_eigen.hpp>
#include <uni20/linalg/decomposition_precision.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 <KernelBackendSelector BackendSelector, uni20::MutableRankedTensorView<2> MatrixTensor, class EigenScalar,
          uni20::MutableRankedTensorView<2> RightEigenvectorTensor>
void nonsymmetric_eigen(BackendSelector&& selector, MatrixTensor&& matrix_work, std::span<EigenScalar> eigenvalues,
                        RightEigenvectorTensor&& right_eigenvectors, bool compute_right_vectors,
                        std::optional<Precision> requested = {})
{
  auto const p = detail::decomposition_precision(matrix_work, requested);
  auto matrix_descriptor = uni20::mdspec_of(matrix_work);
  auto right_eigenvector_descriptor = uni20::mdspec_of(right_eigenvectors);
  detail::dispatch_at_precision<uni20::tensor_element_t<MatrixTensor>>(
      std::forward<BackendSelector>(selector), nonsymmetric_eigen_op{.compute_right_vectors = compute_right_vectors}, p,
      matrix_descriptor, eigenvalues, right_eigenvector_descriptor);
 
  uni20::detail::record_output_precision(matrix_work, p);
  if (compute_right_vectors) uni20::detail::record_output_precision(right_eigenvectors, p);
}
 
template <uni20::MutableRankedTensorView<2> MatrixTensor, class EigenScalar,
          uni20::MutableRankedTensorView<2> RightEigenvectorTensor>
void nonsymmetric_eigen(MatrixTensor&& matrix_work, std::span<EigenScalar> eigenvalues,
                        RightEigenvectorTensor&& right_eigenvectors, bool compute_right_vectors,
                        std::optional<Precision> requested = {})
{
  auto const p = detail::decomposition_precision(matrix_work, requested);
  auto operation = nonsymmetric_eigen_op{.compute_right_vectors = compute_right_vectors};
  auto selector = select_backend(operation, matrix_work, right_eigenvectors);
  nonsymmetric_eigen(selector, std::forward<MatrixTensor>(matrix_work), eigenvalues,
                     std::forward<RightEigenvectorTensor>(right_eigenvectors), compute_right_vectors, p);
}
 
} // namespace uni20::linalg