uni20/ops_2matrix__set_8hpp_source.html

#pragma once


#include <uni20/linalg/backends/cpu/matrix_set.hpp>

#include <uni20/linalg/dispatch.hpp>

#include <uni20/linalg/operation_tags.hpp>

#include <uni20/tensor/concepts.hpp>


#include <utility>


namespace uni20::linalg

{


template <class BackendSelector, uni20::MutableRankedTensorView<2> MatrixTensor, class Scalar>


void set_matrix(BackendSelector&& selector, MatrixTensor&& matrix, Scalar diagonal, Scalar off_diagonal,

                MatrixRegion region = MatrixRegion::All)

{

  auto matrix_descriptor = uni20::mdspec_of(matrix);

  dispatch_kernel(std::forward<BackendSelector>(selector), matrix_set_op{.region = region}, matrix_descriptor, diagonal,

                  off_diagonal);

}


template <uni20::MutableRankedTensorView<2> MatrixTensor, class Scalar>


void set_matrix(MatrixTensor&& matrix, Scalar diagonal, Scalar off_diagonal, MatrixRegion region = MatrixRegion::All)

{

  auto selector = select_backend(matrix_set_op{.region = region}, matrix);

  set_matrix(selector, std::forward<MatrixTensor>(matrix), diagonal, off_diagonal, region);

}


} // namespace uni20::linalg

matrix_set.hpp
Reference CPU backend for structured dense matrix initialization.

uni20::Scalar
Concept for numeric scalar types (integer, real, or complex).
Definition scalar_concepts.hpp:81

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

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

uni20::linalg::set_matrix
void set_matrix(BackendSelector selector, async::Async< MatrixTensor > &matrix, Diagonal &&diagonal, OffDiagonal &&off_diagonal, MatrixRegion region=MatrixRegion::All)
Schedule structured matrix initialization through an explicit selector.
Definition async/matrix_set.hpp:58

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::MatrixRegion
MatrixRegion
Matrix region selected by structured initialization operations.
Definition operation_tags.hpp:208

uni20::linalg::MatrixRegion::All
@ All
All entries, including rectangular portions of the matrix.

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::matrix_set_op
Structured dense matrix initialization operation tag.
Definition operation_tags.hpp:219

uni20::linalg::matrix_set_op::region
MatrixRegion region
Definition operation_tags.hpp:221

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