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

• ck
• tensor_operation
• device
• GemmGemmPadder

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__ constexpr __device__ auto	PadADescriptor_M_K (const ADesc_MRaw_KRaw &a_desc_mraw_kraw) const
template<typename BDesc_NRaw_KRaw >
__host__ constexpr __device__ auto	PadBDescriptor_N_K (const BDesc_NRaw_KRaw &b_desc_nraw_kraw) const
template<typename B1Desc_NRaw_KRaw >
__host__ constexpr __device__ auto	PadB1Descriptor_N_K (const B1Desc_NRaw_KRaw &b1_desc_nraw_kraw) const
template<typename CDesc_MRaw_NRaw >
__host__ constexpr __device__ 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__ constexpr __device__ 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__ constexpr __device__ 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__ constexpr __device__ 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__ constexpr __device__ 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 >

constexpr 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

ck::tensor_operation::device::GemmSpecialization::NKOPadding

@ NKOPadding

ck::tensor_operation::device::GemmSpecialization::KOPadding

@ KOPadding

ck::tensor_operation::device::GemmSpecialization::KPadding

@ KPadding

ck::tensor_operation::device::GemmSpecialization::MNKOPadding

@ MNKOPadding

ck::tensor_operation::device::GemmSpecialization::MNKPadding

@ MNKPadding

ck::tensor_operation::device::GemmSpecialization::MKOPadding

@ MKOPadding

ck::tensor_operation::device::GemmSpecialization::NKPadding

@ NKPadding

◆ PadM

template<GemmSpecialization GemmSpec, typename MPerTileType , typename NPerTileType , typename KPerTileType , typename OPerTileType >

constexpr 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

ck::tensor_operation::device::GemmSpecialization::MOPadding

@ MOPadding

ck::tensor_operation::device::GemmSpecialization::MPadding

@ MPadding

ck::tensor_operation::device::GemmSpecialization::MNPadding

@ MNPadding

◆ PadN

template<GemmSpecialization GemmSpec, typename MPerTileType , typename NPerTileType , typename KPerTileType , typename OPerTileType >

constexpr 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

ck::tensor_operation::device::GemmSpecialization::NPadding

@ NPadding

◆ PadO

template<GemmSpecialization GemmSpec, typename MPerTileType , typename NPerTileType , typename KPerTileType , typename OPerTileType >

constexpr 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

---

The documentation for this struct was generated from the following file:

• /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck/tensor_operation/gpu/device/matrix_padder.hpp