FixedGemmParams Struct Reference

FixedGemmParams Struct Reference#

Composable Kernel: ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_, NumGroupsToMerge_, EnableSplitImage_ >::FixedGemmParams Struct Reference
Public Types | Static Public Attributes | List of all members
ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_, NumGroupsToMerge_, EnableSplitImage_ >::FixedGemmParams Struct Reference

#include <grouped_convolution_utils.hpp>

Public Types

using ELayout = ck_tile::tensor_layout::gemm::RowMajor

Static Public Attributes

static constexpr ck_tile::index_t TilePartitionerGroupNum = 8
static constexpr ck_tile::index_t TilePartitionerM01 = 4
static constexpr bool kPadM = true
static constexpr bool kPadN = true
static constexpr bool kPadK = true
static constexpr bool TransposeC = false
static constexpr bool FixedVectorSize = true
static constexpr bool UseStructuredSparsity = false
static constexpr bool Persistent = false

Member Typedef Documentation

◆ ELayout

template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_, typename WeiLayout_, typename DsLayout_, typename OutLayout_, index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1, index_t NumGroupsToMerge_ = 1, bool EnableSplitImage_ = false>
using ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_, NumGroupsToMerge_, EnableSplitImage_ >::FixedGemmParams::ELayout = ck_tile::tensor_layout::gemm::RowMajor

Member Data Documentation

◆ FixedVectorSize

template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_, typename WeiLayout_, typename DsLayout_, typename OutLayout_, index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1, index_t NumGroupsToMerge_ = 1, bool EnableSplitImage_ = false>
bool ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_, NumGroupsToMerge_, EnableSplitImage_ >::FixedGemmParams::FixedVectorSize = true
staticconstexpr

◆ kPadK

template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_, typename WeiLayout_, typename DsLayout_, typename OutLayout_, index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1, index_t NumGroupsToMerge_ = 1, bool EnableSplitImage_ = false>
bool ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_, NumGroupsToMerge_, EnableSplitImage_ >::FixedGemmParams::kPadK = true
staticconstexpr

◆ kPadM

template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_, typename WeiLayout_, typename DsLayout_, typename OutLayout_, index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1, index_t NumGroupsToMerge_ = 1, bool EnableSplitImage_ = false>
bool ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_, NumGroupsToMerge_, EnableSplitImage_ >::FixedGemmParams::kPadM = true
staticconstexpr

◆ kPadN

template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_, typename WeiLayout_, typename DsLayout_, typename OutLayout_, index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1, index_t NumGroupsToMerge_ = 1, bool EnableSplitImage_ = false>
bool ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_, NumGroupsToMerge_, EnableSplitImage_ >::FixedGemmParams::kPadN = true
staticconstexpr

◆ Persistent

template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_, typename WeiLayout_, typename DsLayout_, typename OutLayout_, index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1, index_t NumGroupsToMerge_ = 1, bool EnableSplitImage_ = false>
bool ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_, NumGroupsToMerge_, EnableSplitImage_ >::FixedGemmParams::Persistent = false
staticconstexpr

◆ TilePartitionerGroupNum

template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_, typename WeiLayout_, typename DsLayout_, typename OutLayout_, index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1, index_t NumGroupsToMerge_ = 1, bool EnableSplitImage_ = false>
ck_tile::index_t ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_, NumGroupsToMerge_, EnableSplitImage_ >::FixedGemmParams::TilePartitionerGroupNum = 8
staticconstexpr

◆ TilePartitionerM01

template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_, typename WeiLayout_, typename DsLayout_, typename OutLayout_, index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1, index_t NumGroupsToMerge_ = 1, bool EnableSplitImage_ = false>
ck_tile::index_t ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_, NumGroupsToMerge_, EnableSplitImage_ >::FixedGemmParams::TilePartitionerM01 = 4
staticconstexpr

◆ TransposeC

template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_, typename WeiLayout_, typename DsLayout_, typename OutLayout_, index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1, index_t NumGroupsToMerge_ = 1, bool EnableSplitImage_ = false>
bool ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_, NumGroupsToMerge_, EnableSplitImage_ >::FixedGemmParams::TransposeC = false
staticconstexpr

◆ UseStructuredSparsity

template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_, typename WeiLayout_, typename DsLayout_, typename OutLayout_, index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1, index_t NumGroupsToMerge_ = 1, bool EnableSplitImage_ = false>
bool ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_, NumGroupsToMerge_, EnableSplitImage_ >::FixedGemmParams::UseStructuredSparsity = false
staticconstexpr
The documentation for this struct was generated from the following file: