gemm-diag.tpp File Reference

#include <batmat/assume.hpp>
#include <batmat/linalg/micro-kernels/gemm-diag.hpp>
#include <batmat/linalg/uview.hpp>
#include <batmat/loop.hpp>
#include <batmat/ops/rotate.hpp>

Go to the source code of this file.

Namespaces
namespace	batmat
namespace	batmat::linalg
namespace	batmat::linalg::micro_kernels
namespace	batmat::linalg::micro_kernels::gemm_diag

Macros
#define	UNROLL_FOR(...)

Functions
template<class T, class Abi, KernelConfig Conf, index_t RowsReg, index_t ColsReg, StorageOrder OA, StorageOrder OB, StorageOrder OC, StorageOrder OD>
std::conditional_t< Conf.track_zeros, std::pair< index_t, index_t >, void >	batmat::linalg::micro_kernels::gemm_diag::gemm_diag_copy_microkernel (const uview< const T, Abi, OA > A, const uview< const T, Abi, OB > B, const std::optional< uview< const T, Abi, OC > > C, const uview< T, Abi, OD > D, const uview_vec< const T, Abi > d, const index_t k) noexcept
	Generalized matrix multiplication D = C ± A⁽ᵀ⁾ diag(d) B⁽ᵀ⁾. Single register block.
template<class T, class Abi, KernelConfig Conf, StorageOrder OA, StorageOrder OB, StorageOrder OC, StorageOrder OD>
void	batmat::linalg::micro_kernels::gemm_diag::gemm_diag_copy_register (const view< const T, Abi, OA > A, const view< const T, Abi, OB > B, const std::optional< view< const T, Abi, OC > > C, const view< T, Abi, OD > D, view< const T, Abi > d) noexcept
	Generalized matrix multiplication D = C ± A⁽ᵀ⁾ diag(d) B⁽ᵀ⁾. Using register blocking.

Variables
template<MatrixStructure Struc>
constexpr auto	batmat::linalg::micro_kernels::gemm_diag::first_column
template<index_t ColsReg, MatrixStructure Struc>
constexpr auto	batmat::linalg::micro_kernels::gemm_diag::last_column

Macro Definition Documentation

UNROLL_FOR

#define UNROLL_FOR

(

...

)

Value:

BATMAT_FULLY_UNROLLED_FOR (__VA_ARGS__)

Definition at line 9 of file gemm-diag.tpp.