gemv.hpp

Go to the documentation of this file.

#pragma once
 
#include <batmat/linalg/structure.hpp>
#include <batmat/linalg/uview.hpp>
#include <batmat/lut.hpp>
#include <batmat/platform/platform.hpp>
#include <optional>
 
namespace batmat::linalg::micro_kernels::gemv {
 
struct KernelConfig {
    bool negate  = false;
    int shift_A  = 0;
    int rotate_B = 0;
    int rotate_C = 0;
    int rotate_D = rotate_C;
    int mask_D   = rotate_D;
};
 
template <class T, class Abi, KernelConfig Conf, index_t RowsReg, StorageOrder OA>
void gemv_copy_microkernel(uview<const T, Abi, OA> A, uview<const T, Abi, StorageOrder::ColMajor> B,
                           std::optional<uview<const T, Abi, StorageOrder::ColMajor>> C,
                           uview<T, Abi, StorageOrder::ColMajor> D, index_t k) noexcept;
 
template <class T, class Abi, KernelConfig Conf, StorageOrder OA>
void gemv_copy_register(view<const T, Abi, OA> A, view<const T, Abi> B,
                        std::optional<view<const T, Abi>> C, view<T, Abi> D) noexcept;
 
template <class T, class Abi>
constexpr index_t RowsReg = 2 * gemm::RowsReg<T, Abi>;
 
template <class T, class Abi, KernelConfig Conf, StorageOrder OA>
inline const constinit auto gemv_copy_lut =
    make_1d_lut<RowsReg<T, Abi>>([]<index_t Row>(index_constant<Row>) {
        return gemv_copy_microkernel<T, Abi, Conf, Row + 1, OA>;
    });
 
} // namespace batmat::linalg::micro_kernels::gemv