gemv.hpp File Reference

Async Tensor wrapper for fixed-output matrix-vector multiplication. More...

#include <uni20/async/async.hpp>
#include <uni20/async/awaiters.hpp>
#include <uni20/async/debug_scheduler.hpp>
#include <uni20/common/trace.hpp>
#include <uni20/linalg/async/concepts.hpp>
#include <uni20/linalg/async/detail/output.hpp>
#include <uni20/linalg/async/dispatch.hpp>
#include <uni20/linalg/ops/gemv.hpp>
#include <memory>
#include <tuple>
#include <utility>

Include dependency graph for async/gemv.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces
namespace	uni20
namespace	uni20::linalg

Functions
template<uni20::AsyncTensorOutput OutputTensor, class Alpha , uni20::RankedTensorView< 2 > MatrixTensor, uni20::RankedTensorView< 1 > InputTensor, class Beta > requires uni20::MutableRankedTensorView<OutputTensor, 1> && AsyncOperationScalar<Alpha, uni20::tensor_element_t<OutputTensor>> && AsyncOperationScalar<Beta, uni20::tensor_element_t<OutputTensor>>
void	uni20::linalg::gemv (async::Async< OutputTensor > &output, Alpha &&alpha, async::Async< MatrixTensor > const &matrix, async::Async< InputTensor > const &input, Beta &&beta)
	Schedule fixed-output GEMV using static Tensor policy.
template<KernelBackendSelector BackendSelector, uni20::AsyncTensorOutput OutputTensor, class Alpha , uni20::RankedTensorView< 2 > MatrixTensor, uni20::RankedTensorView< 1 > InputTensor, class Beta > requires uni20::MutableRankedTensorView<OutputTensor, 1> && AsyncOperationScalar<Alpha, uni20::tensor_element_t<OutputTensor>> && AsyncOperationScalar<Beta, uni20::tensor_element_t<OutputTensor>>
void	uni20::linalg::gemv (BackendSelector selector, async::Async< OutputTensor > &output, Alpha &&alpha, async::Async< MatrixTensor > const &matrix, async::Async< InputTensor > const &input, Beta &&beta)
	Schedule fixed-output GEMV through an explicit backend selector.

Detailed Description

Async Tensor wrapper for fixed-output matrix-vector multiplication.