gemm-diag.hpp

Go to the documentation of this file.

#pragma once
 
#include <batmat/linalg/structure.hpp>
#include <batmat/linalg/uview.hpp>
#include <batmat/micro-kernels/gemm-diag/export.h>
#include <batmat/platform/platform.hpp>
#include <optional>
#include <type_traits>
#include <utility>
 
namespace batmat::linalg::micro_kernels::gemm_diag {
 
struct BATMAT_LINALG_GEMM_DIAG_EXPORT KernelConfig {
    bool negate             = false;
    bool track_zeros        = false;
    MatrixStructure struc_C = MatrixStructure::General;
};
 
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>
gemm_diag_copy_microkernel(uview<const T, Abi, OA> A, uview<const T, Abi, OB> B,
                           std::optional<uview<const T, Abi, OC>> C, uview<T, Abi, OD> D,
                           uview_vec<const T, Abi> diag, index_t k) noexcept;
 
template <class T, class Abi, KernelConfig Conf, StorageOrder OA, StorageOrder OB, StorageOrder OC,
          StorageOrder OD>
BATMAT_LINALG_GEMM_DIAG_EXPORT void
gemm_diag_copy_register(view<const T, Abi, OA> A, view<const T, Abi, OB> B,
                        std::optional<view<const T, Abi, OC>> C, view<T, Abi, OD> D,
                        view<const T, Abi> diag) noexcept;
 
// Square block sizes greatly simplify handling of triangular matrices.
using gemm::RowsReg;
template <class T, class Abi>
constexpr index_t ColsReg = RowsReg<T, Abi>;
 
} // namespace batmat::linalg::micro_kernels::gemm_diag