contract.hpp File Reference

Residual-axis looped GEMM backend for pairwise tensor contraction. More...

#include <uni20/core/scalar_concepts.hpp>
#include <uni20/linalg/backend_selector.hpp>
#include <uni20/linalg/contraction_strides.hpp>
#include <uni20/linalg/dispatch.hpp>
#include <uni20/linalg/operation_tags.hpp>
#include <uni20/linalg/ops/gemm.hpp>
#include <concepts>
#include <string_view>
#include <type_traits>
#include <utility>

Include dependency graph for linalg/backends/looped_gemm/contract.hpp:

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

Go to the source code of this file.

Classes
struct	uni20::linalg::LoopedGemmContractionBackend< GemmSelector >
	Contract through a loop of GEMMs selected from one retained execution selector. More...

Namespaces
namespace	uni20
namespace	uni20::linalg

Functions
template<class GemmSelector , std::size_t LhsRank, std::size_t RhsRank, std::size_t ContractedRank, MutableRankedContractionMdspecLike< ContractionAxes< LhsRank, RhsRank, ContractedRank >::output_rank > OutputMdspec, uni20::Scalar Scalar, RankedContractionMdspecLike< LhsRank > LhsMdspec, RankedContractionMdspecLike< RhsRank > RhsMdspec>
consteval auto	uni20::linalg::kernel_accepts_types (LoopedGemmContractionBackend< GemmSelector > const &, contract_op< LhsRank, RhsRank, ContractedRank > const &, OutputMdspec &, Scalar const &, LhsMdspec &, RhsMdspec &, Scalar const &)
	Report looped-GEMM contraction eligibility for the retained GEMM selector.
template<class GemmSelector >
	uni20::linalg::LoopedGemmContractionBackend (GemmSelector) -> LoopedGemmContractionBackend< std::remove_cvref_t< GemmSelector > >
template<class GemmSelector , std::size_t LhsRank, std::size_t RhsRank, std::size_t ContractedRank, MutableRankedContractionMdspecLike< ContractionAxes< LhsRank, RhsRank, ContractedRank >::output_rank > OutputMdspec, uni20::Scalar Scalar, RankedContractionMdspecLike< LhsRank > LhsMdspec, RankedContractionMdspecLike< RhsRank > RhsMdspec> requires (detail::looped_gemm_contraction_backend::gemm_dispatch_types_compatible<GemmSelector, OutputMdspec, Scalar, LhsMdspec, RhsMdspec>())
KernelAttempt	uni20::linalg::try_kernel (LoopedGemmContractionBackend< GemmSelector > const &backend, contract_op< LhsRank, RhsRank, ContractedRank > const &operation, OutputMdspec &output, Scalar alpha, LhsMdspec &lhs, RhsMdspec &rhs, Scalar beta)
	Execute one residual M or N loop through nested GEMM dispatch.

Detailed Description

Residual-axis looped GEMM backend for pairwise tensor contraction.