/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/tensor_operation/gpu/device/matrix_padder.hpp Source File#
matrix_padder.hpp
Go to the documentation of this file.
__host__ constexpr __device__ auto integer_divide_ceil(X x, Y y)
Definition: math.hpp:72
auto grid_desc(MatrixPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType > matrix_padder, CDesc_MRaw_NRaw conv_desc)
Definition: matrix_padder.hpp:190
__host__ constexpr __device__ auto PadTensorDescriptor(const TensorDesc &desc, const TileLengths &tile_lengths, DoPads)
Definition: matrix_padder.hpp:19
GemmSpecialization
Definition: gemm_specialization.hpp:11
Definition: ck.hpp:266
__host__ constexpr __device__ auto generate_tuple(F &&f, Number< N >)
Definition: tuple_helper.hpp:21
__host__ constexpr __device__ auto make_pass_through_transform(const LowLength &low_length)
Definition: multi_index_transform_helper.hpp:12
__host__ constexpr __device__ auto transform_tensor_descriptor(const OldTensorDescriptor &old_tensor_desc, const NewTransforms &new_transforms, NewLowerDimensionOldVisibleIdss, NewUpperDimensionNewVisibleIdss)
Definition: tensor_descriptor.hpp:319
__host__ constexpr __device__ auto make_right_pad_transform(const LowLength &low_length, const RightPadLength &right_pad, integral_constant< bool, SkipIsValidCheck >=integral_constant< bool, false >{})
Definition: multi_index_transform_helper.hpp:37
Definition: sequence.hpp:43
Definition: integral_constant.hpp:20
Definition: matrix_padder.hpp:63
KPerTileType KPerTile_
Definition: matrix_padder.hpp:124
OPerTileType OPerTile_
Definition: matrix_padder.hpp:125
static constexpr bool PadM
Definition: matrix_padder.hpp:65
MPerTileType MPerTile_
Definition: matrix_padder.hpp:122
static constexpr bool PadN
Definition: matrix_padder.hpp:70
__host__ constexpr __device__ auto PadCDescriptor_M_N(const CDesc_MRaw_NRaw &c_desc_mraw_nraw) const
Definition: matrix_padder.hpp:116
__host__ constexpr __device__ auto PadADescriptor_M_K(const ADesc_MRaw_KRaw &a_desc_mraw_kraw) const
Definition: matrix_padder.hpp:89
NPerTileType NPerTile_
Definition: matrix_padder.hpp:123
static constexpr bool PadO
Definition: matrix_padder.hpp:80
__host__ constexpr __device__ auto PadB1Descriptor_N_K(const B1Desc_NRaw_KRaw &b1_desc_nraw_kraw) const
Definition: matrix_padder.hpp:107
static constexpr bool PadK
Definition: matrix_padder.hpp:75
__host__ constexpr __device__ auto PadBDescriptor_N_K(const BDesc_NRaw_KRaw &b_desc_nraw_kraw) const
Definition: matrix_padder.hpp:98
Definition: matrix_padder.hpp:204
__host__ constexpr __device__ auto PadADescriptor_M_K(const ADesc_MRaw_KRaw &a_desc_mraw_kraw) const
Definition: matrix_padder.hpp:207
MPerTileType MPerTile_
Definition: matrix_padder.hpp:229
NPerTileType NPerTile_
Definition: matrix_padder.hpp:230
__host__ constexpr __device__ auto PadBDescriptor_N_K(const BDesc_NRaw_KRaw &b_desc_nraw_kraw) const
Definition: matrix_padder.hpp:215
__host__ constexpr __device__ auto PadCDescriptor_M_N(const CDesc_MRaw_NRaw &c_desc_mraw_nraw) const
Definition: matrix_padder.hpp:223
KPerTileType KPerTile_
Definition: matrix_padder.hpp:231
Definition: matrix_padder.hpp:134
NPerTileType NPerTile_
Definition: matrix_padder.hpp:170
MPerTileType MPerTile_
Definition: matrix_padder.hpp:169
static constexpr bool PadK
Definition: matrix_padder.hpp:141
KPerTileType KPerTile_
Definition: matrix_padder.hpp:171
__host__ constexpr __device__ auto PadADescriptor_M_K(const ADesc_MRaw_KRaw &a_desc_mraw_kraw) const
Definition: matrix_padder.hpp:147
__host__ constexpr __device__ auto PadBDescriptor_N_K(const BDesc_NRaw_KRaw &b_desc_nraw_kraw) const
Definition: matrix_padder.hpp:155
__host__ constexpr __device__ auto PadCDescriptor_M_N(const CDesc_MRaw_NRaw &c_desc_mraw_nraw) const
Definition: matrix_padder.hpp:163
static constexpr bool PadM
Definition: matrix_padder.hpp:135
static constexpr bool PadN
Definition: matrix_padder.hpp:138
Definition: matrix_padder.hpp:242
KPerTileType KPerTile_
Definition: matrix_padder.hpp:391
__host__ constexpr __device__ auto PadBDescriptor_N_K(const BDesc_NRaw_KRaw &b_desc_nraw_kraw) const
Definition: matrix_padder.hpp:297
static constexpr auto I2
Definition: matrix_padder.hpp:245
MPerTileType MPerTile_
Definition: matrix_padder.hpp:389
__host__ constexpr __device__ auto PadCDescriptor_M_N(const CDesc_MRaw_NRaw &c_desc_mraw_nraw) const
Definition: matrix_padder.hpp:344
static constexpr auto I3
Definition: matrix_padder.hpp:246
static constexpr auto I0
Definition: matrix_padder.hpp:243
NPerTileType NPerTile_
Definition: matrix_padder.hpp:390
__host__ constexpr __device__ auto PadADescriptor_M_K(const ADesc_MRaw_KRaw &a_desc_mraw_kraw) const
Definition: matrix_padder.hpp:250
static constexpr auto I1
Definition: matrix_padder.hpp:244
Definition: matrix_padder.hpp:180