GemmGemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType, OPerTileType > Struct Template Reference

GemmGemmPadder&lt; GemmSpec, MPerTileType, NPerTileType, KPerTileType, OPerTileType &gt; Struct Template Reference#

Composable Kernel: ck::tensor_operation::device::GemmGemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType, OPerTileType > Struct Template Reference
Public Member Functions | Public Attributes | Static Public Attributes | List of all members
ck::tensor_operation::device::GemmGemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType, OPerTileType > Struct Template Reference

#include <matrix_padder.hpp>

Public Member Functions

template<typename ADesc_MRaw_KRaw>
__host__ __device__ constexpr auto PadADescriptor_M_K (const ADesc_MRaw_KRaw &a_desc_mraw_kraw) const
template<typename BDesc_NRaw_KRaw>
__host__ __device__ constexpr auto PadBDescriptor_N_K (const BDesc_NRaw_KRaw &b_desc_nraw_kraw) const
template<typename B1Desc_NRaw_KRaw>
__host__ __device__ constexpr auto PadB1Descriptor_N_K (const B1Desc_NRaw_KRaw &b1_desc_nraw_kraw) const
template<typename CDesc_MRaw_NRaw>
__host__ __device__ constexpr auto PadCDescriptor_M_N (const CDesc_MRaw_NRaw &c_desc_mraw_nraw) const

Public Attributes

MPerTileType MPerTile_
NPerTileType NPerTile_
KPerTileType KPerTile_
OPerTileType OPerTile_

Static Public Attributes

static constexpr bool PadM
static constexpr bool PadN
static constexpr bool PadK
static constexpr bool PadO

Member Function Documentation

◆ PadADescriptor_M_K()

template<GemmSpecialization GemmSpec, typename MPerTileType, typename NPerTileType, typename KPerTileType, typename OPerTileType>
template<typename ADesc_MRaw_KRaw>
__host__ __device__ constexpr auto ck::tensor_operation::device::GemmGemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType, OPerTileType >::PadADescriptor_M_K ( const ADesc_MRaw_KRaw & a_desc_mraw_kraw) const
inlineconstexpr

◆ PadB1Descriptor_N_K()

template<GemmSpecialization GemmSpec, typename MPerTileType, typename NPerTileType, typename KPerTileType, typename OPerTileType>
template<typename B1Desc_NRaw_KRaw>
__host__ __device__ constexpr auto ck::tensor_operation::device::GemmGemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType, OPerTileType >::PadB1Descriptor_N_K ( const B1Desc_NRaw_KRaw & b1_desc_nraw_kraw) const
inlineconstexpr

◆ PadBDescriptor_N_K()

template<GemmSpecialization GemmSpec, typename MPerTileType, typename NPerTileType, typename KPerTileType, typename OPerTileType>
template<typename BDesc_NRaw_KRaw>
__host__ __device__ constexpr auto ck::tensor_operation::device::GemmGemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType, OPerTileType >::PadBDescriptor_N_K ( const BDesc_NRaw_KRaw & b_desc_nraw_kraw) const
inlineconstexpr

◆ PadCDescriptor_M_N()

template<GemmSpecialization GemmSpec, typename MPerTileType, typename NPerTileType, typename KPerTileType, typename OPerTileType>
template<typename CDesc_MRaw_NRaw>
__host__ __device__ constexpr auto ck::tensor_operation::device::GemmGemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType, OPerTileType >::PadCDescriptor_M_N ( const CDesc_MRaw_NRaw & c_desc_mraw_nraw) const
inlineconstexpr

Member Data Documentation

◆ KPerTile_

template<GemmSpecialization GemmSpec, typename MPerTileType, typename NPerTileType, typename KPerTileType, typename OPerTileType>
KPerTileType ck::tensor_operation::device::GemmGemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType, OPerTileType >::KPerTile_

◆ MPerTile_

template<GemmSpecialization GemmSpec, typename MPerTileType, typename NPerTileType, typename KPerTileType, typename OPerTileType>
MPerTileType ck::tensor_operation::device::GemmGemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType, OPerTileType >::MPerTile_

◆ NPerTile_

template<GemmSpecialization GemmSpec, typename MPerTileType, typename NPerTileType, typename KPerTileType, typename OPerTileType>
NPerTileType ck::tensor_operation::device::GemmGemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType, OPerTileType >::NPerTile_

◆ OPerTile_

template<GemmSpecialization GemmSpec, typename MPerTileType, typename NPerTileType, typename KPerTileType, typename OPerTileType>
OPerTileType ck::tensor_operation::device::GemmGemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType, OPerTileType >::OPerTile_

◆ PadK

template<GemmSpecialization GemmSpec, typename MPerTileType, typename NPerTileType, typename KPerTileType, typename OPerTileType>
bool ck::tensor_operation::device::GemmGemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType, OPerTileType >::PadK
staticconstexpr
Initial value:
=
GemmSpec == GemmSpecialization::KPadding || GemmSpec == GemmSpecialization::MKPadding ||
GemmSpec == GemmSpecialization::NKPadding || GemmSpec == GemmSpecialization::MNKPadding ||
GemmSpec == GemmSpecialization::KOPadding || GemmSpec == GemmSpecialization::MKOPadding ||
GemmSpec == GemmSpecialization::NKOPadding || GemmSpec == GemmSpecialization::MNKOPadding
ck::tensor_operation::device::GemmSpecialization::MKPadding
@ MKPadding
Definition gemm_specialization.hpp:18
ck::tensor_operation::device::GemmSpecialization::NKOPadding
@ NKOPadding
Definition gemm_specialization.hpp:28
ck::tensor_operation::device::GemmSpecialization::KOPadding
@ KOPadding
Definition gemm_specialization.hpp:25
ck::tensor_operation::device::GemmSpecialization::KPadding
@ KPadding
Definition gemm_specialization.hpp:16
ck::tensor_operation::device::GemmSpecialization::MNKOPadding
@ MNKOPadding
Definition gemm_specialization.hpp:29
ck::tensor_operation::device::GemmSpecialization::MNKPadding
@ MNKPadding
Definition gemm_specialization.hpp:20
ck::tensor_operation::device::GemmSpecialization::MKOPadding
@ MKOPadding
Definition gemm_specialization.hpp:27
ck::tensor_operation::device::GemmSpecialization::NKPadding
@ NKPadding
Definition gemm_specialization.hpp:19

◆ PadM

template<GemmSpecialization GemmSpec, typename MPerTileType, typename NPerTileType, typename KPerTileType, typename OPerTileType>
bool ck::tensor_operation::device::GemmGemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType, OPerTileType >::PadM
staticconstexpr
Initial value:
=
GemmSpec == GemmSpecialization::MPadding || GemmSpec == GemmSpecialization::MNPadding ||
GemmSpec == GemmSpecialization::MKPadding || GemmSpec == GemmSpecialization::MNKPadding ||
GemmSpec == GemmSpecialization::MOPadding || GemmSpec == GemmSpecialization::MNOPadding ||
GemmSpec == GemmSpecialization::MKOPadding || GemmSpec == GemmSpecialization::MNKOPadding
ck::tensor_operation::device::GemmSpecialization::MNOPadding
@ MNOPadding
Definition gemm_specialization.hpp:26
ck::tensor_operation::device::GemmSpecialization::MOPadding
@ MOPadding
Definition gemm_specialization.hpp:23
ck::tensor_operation::device::GemmSpecialization::MPadding
@ MPadding
Definition gemm_specialization.hpp:14
ck::tensor_operation::device::GemmSpecialization::MNPadding
@ MNPadding
Definition gemm_specialization.hpp:17

◆ PadN

template<GemmSpecialization GemmSpec, typename MPerTileType, typename NPerTileType, typename KPerTileType, typename OPerTileType>
bool ck::tensor_operation::device::GemmGemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType, OPerTileType >::PadN
staticconstexpr
Initial value:
=
GemmSpec == GemmSpecialization::NPadding || GemmSpec == GemmSpecialization::MNPadding ||
GemmSpec == GemmSpecialization::NKPadding || GemmSpec == GemmSpecialization::MNKPadding ||
GemmSpec == GemmSpecialization::NOPadding || GemmSpec == GemmSpecialization::MNOPadding ||
GemmSpec == GemmSpecialization::NKOPadding || GemmSpec == GemmSpecialization::MNKOPadding
ck::tensor_operation::device::GemmSpecialization::NOPadding
@ NOPadding
Definition gemm_specialization.hpp:24
ck::tensor_operation::device::GemmSpecialization::NPadding
@ NPadding
Definition gemm_specialization.hpp:15

◆ PadO

template<GemmSpecialization GemmSpec, typename MPerTileType, typename NPerTileType, typename KPerTileType, typename OPerTileType>
bool ck::tensor_operation::device::GemmGemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType, OPerTileType >::PadO
staticconstexpr
Initial value:
=
GemmSpec == GemmSpecialization::OPadding || GemmSpec == GemmSpecialization::MOPadding ||
GemmSpec == GemmSpecialization::NOPadding || GemmSpec == GemmSpecialization::KOPadding ||
GemmSpec == GemmSpecialization::MNOPadding || GemmSpec == GemmSpecialization::MKOPadding ||
GemmSpec == GemmSpecialization::NKOPadding || GemmSpec == GemmSpecialization::MNKOPadding
ck::tensor_operation::device::GemmSpecialization::OPadding
@ OPadding
Definition gemm_specialization.hpp:22
The documentation for this struct was generated from the following file: