contract.hpp File Reference

#include <uni20/mdspan/mdspan.hpp>
#include <uni20/mdspan/strides.hpp>

Include dependency graph for kernel/cpu/contract.hpp:

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

Go to the source code of this file.

Classes
class	uni20::kernel::cpu::GemmLoop< T, MR, NR, KR >
	Generic M×N×K contraction engine with full K-dimension recursion. More...

Namespaces
namespace	uni20
namespace	uni20::kernel
namespace	uni20::kernel::cpu

Functions
template<typename T , std::size_t MR, std::size_t NR, std::size_t KR>
void	uni20::kernel::contract_strided (static_vector< extent_strides< 2 >, MR > const &Mgrp, static_vector< extent_strides< 2 >, NR > const &Ngrp, static_vector< extent_strides< 2 >, KR > const &Kgrp, T alpha, T const *A, T const *B, T beta, T *C)
	Execute the CPU tensor contraction using precomputed stride groupings.