GemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType > Struct Template Reference

• ck
• tensor_operation
• device
• GemmPadder

Public Member Functions | Public Attributes | Static Public Attributes | List of all members

ck::tensor_operation::device::GemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType > Struct Template Reference

#include <matrix_padder.hpp>

Inheritance diagram for ck::tensor_operation::device::GemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType >:

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 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_

Static Public Attributes
static constexpr bool	PadM
static constexpr bool	PadN
static constexpr bool	PadK

Member Function Documentation

◆ PadADescriptor_M_K()

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

template<typename ADesc_MRaw_KRaw>

__host__ __device__ constexpr auto ck::tensor_operation::device::GemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType >::PadADescriptor_M_K ( const ADesc_MRaw_KRaw & a_desc_mraw_kraw ) const

inlineconstexpr

◆ PadBDescriptor_N_K()

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

template<typename BDesc_NRaw_KRaw>

__host__ __device__ constexpr auto ck::tensor_operation::device::GemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType >::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>

template<typename CDesc_MRaw_NRaw>

__host__ __device__ constexpr auto ck::tensor_operation::device::GemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType >::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>

KPerTileType ck::tensor_operation::device::GemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType >::KPerTile_

◆ MPerTile_

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

MPerTileType ck::tensor_operation::device::GemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType >::MPerTile_

◆ NPerTile_

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

NPerTileType ck::tensor_operation::device::GemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType >::NPerTile_

◆ PadK

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

bool ck::tensor_operation::device::GemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType >::PadK

staticconstexpr

Initial value:

=

(GemmSpec == GemmSpecialization::KPadding || GemmSpec == GemmSpecialization::MKPadding ||

GemmSpec == GemmSpecialization::NKPadding || GemmSpec == GemmSpecialization::MNKPadding)

ck::tensor_operation::device::GemmSpecialization::MKPadding

@ MKPadding

Definition gemm_specialization.hpp:18

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

@ KPadding

Definition gemm_specialization.hpp:16

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

@ MNKPadding

Definition gemm_specialization.hpp:20

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

@ NKPadding

Definition gemm_specialization.hpp:19

◆ PadM

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

bool ck::tensor_operation::device::GemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType >::PadM

staticconstexpr

Initial value:

=

(GemmSpec == GemmSpecialization::MPadding || GemmSpec == GemmSpecialization::MNPadding ||

GemmSpec == GemmSpecialization::MKPadding || GemmSpec == GemmSpecialization::MNKPadding)

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>

bool ck::tensor_operation::device::GemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType >::PadN

staticconstexpr

Initial value:

=

(GemmSpec == GemmSpecialization::NPadding || GemmSpec == GemmSpecialization::MNPadding ||

GemmSpec == GemmSpecialization::NKPadding || GemmSpec == GemmSpecialization::MNKPadding)

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

@ NPadding

Definition gemm_specialization.hpp:15

---

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

• matrix_padder.hpp