uni20/ops_2matrix__exponential_8hpp_source.html

#pragma once


#include <uni20/linalg/backends/cpu/matrix_exponential.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> OutputTensor, uni20::RankedTensorView<2> InputTensor,

          class TimeScalar>


void matrix_exponential(BackendSelector&& selector, OutputTensor&& output, InputTensor const& input, TimeScalar time)

{

  auto output_descriptor = uni20::mdspec_of(output);

  auto input_descriptor = uni20::mdspec_of(input);

  dispatch_kernel(std::forward<BackendSelector>(selector), matrix_exponential_op{}, output_descriptor, input_descriptor,

                  time);

}


template <uni20::MutableRankedTensorView<2> OutputTensor, uni20::RankedTensorView<2> InputTensor, class TimeScalar>


void matrix_exponential(OutputTensor&& output, InputTensor const& input, TimeScalar time)

{

  auto selector = select_backend(matrix_exponential_op{}, output, input);

  matrix_exponential(selector, std::forward<OutputTensor>(output), input, time);

}


} // namespace uni20::linalg

matrix_exponential.hpp
CPU dense matrix exponential interface and dispatch adapter.

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::matrix_exponential
void matrix_exponential(BackendSelector selector, async::Async< OutputTensor > &output, async::Async< InputTensor > const &input, TimeScalar &&time)
Schedule fixed-output matrix exponentiation through an explicit selector.
Definition async/matrix_exponential.hpp:64

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_exponential_op
Dense matrix exponential operation tag.
Definition operation_tags.hpp:202

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