Uni20
Asynchronous tensor and runtime library.
Loading...
Searching...
No Matches
Namespaces | Concepts | Functions
gemm.hpp File Reference

Accessor-respecting reference CPU GEMM for resolved mdspans. More...

#include <uni20/common/trace.hpp>
#include <uni20/core/runtime_precision.hpp>
#include <uni20/core/scalar_concepts.hpp>
#include <uni20/mdspan/concepts.hpp>
#include <concepts>
#include <type_traits>
Include dependency graph for linalg/cpu/gemm.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
 
namespace  uni20::linalg::cpu
 

Concepts

concept  uni20::linalg::cpu::GemmCompatible
 Report whether resolved mdspans support reference CPU GEMM expressions.
 

Functions

template<class OutputMdspan , uni20::Scalar Scalar, class LhsMdspan , class RhsMdspan >
requires GemmCompatible<OutputMdspan, Scalar, LhsMdspan, RhsMdspan>
void uni20::linalg::cpu::gemm (OutputMdspan &&output, Scalar alpha, LhsMdspan &&lhs, RhsMdspan &&rhs, Scalar beta)
 Execute reference CPU GEMM on resolved mdspans.
 

Detailed Description

Accessor-respecting reference CPU GEMM for resolved mdspans.