batmat 0.0.15
Batched linear algebra routines
Loading...
Searching...
No Matches
potrf.hpp File Reference

Go to the source code of this file.

Classes

struct  batmat::linalg::micro_kernels::potrf::KernelConfig

Namespaces

namespace  batmat
namespace  batmat::linalg
namespace  batmat::linalg::micro_kernels
namespace  batmat::linalg::micro_kernels::potrf

Typedefs

template<class T, class Abi, KernelConfig Conf>
using batmat::linalg::micro_kernels::potrf::diag_uview_type = std::conditional_t<Conf.with_diag(), uview_vec<T, Abi>, std::false_type>
template<class T, class Abi, KernelConfig Conf>
using batmat::linalg::micro_kernels::potrf::diag_view_type = std::conditional_t<Conf.with_diag(), view<T, Abi>, std::false_type>

Functions

template<class T, class Abi, KernelConfig Conf, index_t RowsReg, StorageOrder O1, StorageOrder O2>
void batmat::linalg::micro_kernels::potrf::potrf_copy_microkernel (uview< const T, Abi, O1 > A1, uview< const T, Abi, O2 > A2, uview< const T, Abi, O2 > C, uview< T, Abi, O2 > D, T *invD, index_t k1, index_t k2, T regularization, diag_uview_type< const T, Abi, Conf > diag) noexcept
template<class T, class Abi, KernelConfig Conf, index_t RowsReg, index_t ColsReg, StorageOrder O1, StorageOrder O2>
void batmat::linalg::micro_kernels::potrf::trsm_copy_microkernel (uview< const T, Abi, O1 > A1, uview< const T, Abi, O1 > B1, uview< const T, Abi, O2 > A2, uview< const T, Abi, O2 > B2, uview< const T, Abi, O2 > L, const T *invL, uview< const T, Abi, O2 > C, uview< T, Abi, O2 > D, index_t k1, index_t k2, diag_uview_type< const T, Abi, Conf > diag) noexcept
template<class T, class Abi, KernelConfig Conf, StorageOrder OA, StorageOrder OCD>
void batmat::linalg::micro_kernels::potrf::potrf_copy_register (view< const T, Abi, OA > A, view< const T, Abi, OCD > C, view< T, Abi, OCD > D, T regularization, diag_view_type< const T, Abi, Conf > diag) noexcept

Variables

template<class T, class Abi>
constexpr index_t batmat::linalg::micro_kernels::potrf::ColsReg = RowsReg<T, Abi>
template<class T, class Abi, KernelConfig Conf, StorageOrder OA, StorageOrder OC>
const constinit auto batmat::linalg::micro_kernels::potrf::potrf_copy_lut
template<class T, class Abi, KernelConfig Conf, StorageOrder O1, StorageOrder O2>
const constinit auto batmat::linalg::micro_kernels::potrf::trsm_copy_lut
template<class T, class Abi>
constexpr index_t batmat::linalg::micro_kernels::potrf::RowsReg
 Register block size of the matrix-matrix multiplication micro-kernels.