MoeSortingProblemMp< IndexType_, WeightType_, MeshType_, SubTokenTile_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_ > Struct Template Reference

MoeSortingProblemMp&lt; IndexType_, WeightType_, MeshType_, SubTokenTile_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_ &gt; Struct Template Reference#

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

#include <moe_sorting_problem.hpp>

Public Types

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

Static Public Attributes

static constexpr index_t SubTokenTile = SubTokenTile_
static constexpr bool LocalExpertMasking = LocalExpertMasking_
static constexpr bool LocalToken = LocalToken_
static constexpr bool SkipExpertsWithZeroTokens = SkipExpertsWithZeroTokens_

Member Typedef Documentation

◆ IndexType

template<typename IndexType_, typename WeightType_, typename MeshType_, index_t SubTokenTile_, bool LocalExpertMasking_, bool LocalToken_, bool SkipExpertsWithZeroTokens_ = true>
using ck_tile::MoeSortingProblemMp< IndexType_, WeightType_, MeshType_, SubTokenTile_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_ >::IndexType = remove_cvref_t<IndexType_>

◆ MeshType

template<typename IndexType_, typename WeightType_, typename MeshType_, index_t SubTokenTile_, bool LocalExpertMasking_, bool LocalToken_, bool SkipExpertsWithZeroTokens_ = true>
using ck_tile::MoeSortingProblemMp< IndexType_, WeightType_, MeshType_, SubTokenTile_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_ >::MeshType = remove_cvref_t<MeshType_>

◆ WeightType

template<typename IndexType_, typename WeightType_, typename MeshType_, index_t SubTokenTile_, bool LocalExpertMasking_, bool LocalToken_, bool SkipExpertsWithZeroTokens_ = true>
using ck_tile::MoeSortingProblemMp< IndexType_, WeightType_, MeshType_, SubTokenTile_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_ >::WeightType = remove_cvref_t<WeightType_>

Member Data Documentation

◆ LocalExpertMasking

template<typename IndexType_, typename WeightType_, typename MeshType_, index_t SubTokenTile_, bool LocalExpertMasking_, bool LocalToken_, bool SkipExpertsWithZeroTokens_ = true>
bool ck_tile::MoeSortingProblemMp< IndexType_, WeightType_, MeshType_, SubTokenTile_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_ >::LocalExpertMasking = LocalExpertMasking_
staticconstexpr

◆ LocalToken

template<typename IndexType_, typename WeightType_, typename MeshType_, index_t SubTokenTile_, bool LocalExpertMasking_, bool LocalToken_, bool SkipExpertsWithZeroTokens_ = true>
bool ck_tile::MoeSortingProblemMp< IndexType_, WeightType_, MeshType_, SubTokenTile_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_ >::LocalToken = LocalToken_
staticconstexpr

◆ SkipExpertsWithZeroTokens

template<typename IndexType_, typename WeightType_, typename MeshType_, index_t SubTokenTile_, bool LocalExpertMasking_, bool LocalToken_, bool SkipExpertsWithZeroTokens_ = true>
bool ck_tile::MoeSortingProblemMp< IndexType_, WeightType_, MeshType_, SubTokenTile_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_ >::SkipExpertsWithZeroTokens = SkipExpertsWithZeroTokens_
staticconstexpr

◆ SubTokenTile

template<typename IndexType_, typename WeightType_, typename MeshType_, index_t SubTokenTile_, bool LocalExpertMasking_, bool LocalToken_, bool SkipExpertsWithZeroTokens_ = true>
index_t ck_tile::MoeSortingProblemMp< IndexType_, WeightType_, MeshType_, SubTokenTile_, LocalExpertMasking_, LocalToken_, SkipExpertsWithZeroTokens_ >::SubTokenTile = SubTokenTile_
staticconstexpr
The documentation for this struct was generated from the following file: