MoeSortingProblemEx< IndexType_, WeightType_, SubTokenTile_, SubTokenOneShot_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_, ExpertTile_ > Struct Template Reference

MoeSortingProblemEx&lt; IndexType_, WeightType_, SubTokenTile_, SubTokenOneShot_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_, ExpertTile_ &gt; Struct Template Reference#

Composable Kernel: ck_tile::MoeSortingProblemEx< IndexType_, WeightType_, SubTokenTile_, SubTokenOneShot_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_, ExpertTile_ > Struct Template Reference
Public Types | Static Public Attributes | List of all members
ck_tile::MoeSortingProblemEx< IndexType_, WeightType_, SubTokenTile_, SubTokenOneShot_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_, ExpertTile_ > Struct Template Reference

#include <moe_sorting_problem.hpp>

Public Types

using WeightType = remove_cvref_t<WeightType_>
using IndexType = remove_cvref_t<IndexType_>

Static Public Attributes

static constexpr index_t WarpSize = get_warp_size()
static constexpr index_t WarpsPerBlock = 1
static constexpr index_t SubTokenTile = SubTokenTile_
static constexpr bool SubTokenOneShot = SubTokenOneShot_
static constexpr bool LocalExpertMasking = LocalExpertMasking_
static constexpr bool LocalToken = LocalToken_
static constexpr bool SkipExpertsWithZeroTokens = SkipExpertsWithZeroTokens_
static constexpr index_t ExpertTile = ExpertTile_

Member Typedef Documentation

◆ IndexType

template<typename IndexType_, typename WeightType_, index_t SubTokenTile_, bool SubTokenOneShot_, bool LocalExpertMasking_, bool LocalToken_, bool SkipExpertsWithZeroTokens_ = true, index_t ExpertTile_ = 0>
using ck_tile::MoeSortingProblemEx< IndexType_, WeightType_, SubTokenTile_, SubTokenOneShot_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_, ExpertTile_ >::IndexType = remove_cvref_t<IndexType_>

◆ WeightType

template<typename IndexType_, typename WeightType_, index_t SubTokenTile_, bool SubTokenOneShot_, bool LocalExpertMasking_, bool LocalToken_, bool SkipExpertsWithZeroTokens_ = true, index_t ExpertTile_ = 0>
using ck_tile::MoeSortingProblemEx< IndexType_, WeightType_, SubTokenTile_, SubTokenOneShot_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_, ExpertTile_ >::WeightType = remove_cvref_t<WeightType_>

Member Data Documentation

◆ ExpertTile

template<typename IndexType_, typename WeightType_, index_t SubTokenTile_, bool SubTokenOneShot_, bool LocalExpertMasking_, bool LocalToken_, bool SkipExpertsWithZeroTokens_ = true, index_t ExpertTile_ = 0>
index_t ck_tile::MoeSortingProblemEx< IndexType_, WeightType_, SubTokenTile_, SubTokenOneShot_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_, ExpertTile_ >::ExpertTile = ExpertTile_
staticconstexpr

◆ LocalExpertMasking

template<typename IndexType_, typename WeightType_, index_t SubTokenTile_, bool SubTokenOneShot_, bool LocalExpertMasking_, bool LocalToken_, bool SkipExpertsWithZeroTokens_ = true, index_t ExpertTile_ = 0>
bool ck_tile::MoeSortingProblemEx< IndexType_, WeightType_, SubTokenTile_, SubTokenOneShot_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_, ExpertTile_ >::LocalExpertMasking = LocalExpertMasking_
staticconstexpr

◆ LocalToken

template<typename IndexType_, typename WeightType_, index_t SubTokenTile_, bool SubTokenOneShot_, bool LocalExpertMasking_, bool LocalToken_, bool SkipExpertsWithZeroTokens_ = true, index_t ExpertTile_ = 0>
bool ck_tile::MoeSortingProblemEx< IndexType_, WeightType_, SubTokenTile_, SubTokenOneShot_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_, ExpertTile_ >::LocalToken = LocalToken_
staticconstexpr

◆ SkipExpertsWithZeroTokens

template<typename IndexType_, typename WeightType_, index_t SubTokenTile_, bool SubTokenOneShot_, bool LocalExpertMasking_, bool LocalToken_, bool SkipExpertsWithZeroTokens_ = true, index_t ExpertTile_ = 0>
bool ck_tile::MoeSortingProblemEx< IndexType_, WeightType_, SubTokenTile_, SubTokenOneShot_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_, ExpertTile_ >::SkipExpertsWithZeroTokens = SkipExpertsWithZeroTokens_
staticconstexpr

◆ SubTokenOneShot

template<typename IndexType_, typename WeightType_, index_t SubTokenTile_, bool SubTokenOneShot_, bool LocalExpertMasking_, bool LocalToken_, bool SkipExpertsWithZeroTokens_ = true, index_t ExpertTile_ = 0>
bool ck_tile::MoeSortingProblemEx< IndexType_, WeightType_, SubTokenTile_, SubTokenOneShot_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_, ExpertTile_ >::SubTokenOneShot = SubTokenOneShot_
staticconstexpr

◆ SubTokenTile

template<typename IndexType_, typename WeightType_, index_t SubTokenTile_, bool SubTokenOneShot_, bool LocalExpertMasking_, bool LocalToken_, bool SkipExpertsWithZeroTokens_ = true, index_t ExpertTile_ = 0>
index_t ck_tile::MoeSortingProblemEx< IndexType_, WeightType_, SubTokenTile_, SubTokenOneShot_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_, ExpertTile_ >::SubTokenTile = SubTokenTile_
staticconstexpr

◆ WarpSize

template<typename IndexType_, typename WeightType_, index_t SubTokenTile_, bool SubTokenOneShot_, bool LocalExpertMasking_, bool LocalToken_, bool SkipExpertsWithZeroTokens_ = true, index_t ExpertTile_ = 0>
index_t ck_tile::MoeSortingProblemEx< IndexType_, WeightType_, SubTokenTile_, SubTokenOneShot_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_, ExpertTile_ >::WarpSize = get_warp_size()
staticconstexpr

◆ WarpsPerBlock

template<typename IndexType_, typename WeightType_, index_t SubTokenTile_, bool SubTokenOneShot_, bool LocalExpertMasking_, bool LocalToken_, bool SkipExpertsWithZeroTokens_ = true, index_t ExpertTile_ = 0>
index_t ck_tile::MoeSortingProblemEx< IndexType_, WeightType_, SubTokenTile_, SubTokenOneShot_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_, ExpertTile_ >::WarpsPerBlock = 1
staticconstexpr
The documentation for this struct was generated from the following file: