MoeSortingProblem< IndexType_, WeightType_, InternalLoadUnroll_, ExpertTile_ > Struct Template Reference

MoeSortingProblem&lt; IndexType_, WeightType_, InternalLoadUnroll_, ExpertTile_ &gt; Struct Template Reference#

Composable Kernel: ck_tile::MoeSortingProblem< IndexType_, WeightType_, InternalLoadUnroll_, ExpertTile_ > Struct Template Reference
Public Types | Static Public Attributes | List of all members
ck_tile::MoeSortingProblem< IndexType_, WeightType_, InternalLoadUnroll_, 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 InternalLoadUnroll
static constexpr index_t ExpertTile = ExpertTile_

Member Typedef Documentation

◆ IndexType

template<typename IndexType_, typename WeightType_, index_t InternalLoadUnroll_, index_t ExpertTile_ = 0>
using ck_tile::MoeSortingProblem< IndexType_, WeightType_, InternalLoadUnroll_, ExpertTile_ >::IndexType = remove_cvref_t<IndexType_>

◆ WeightType

template<typename IndexType_, typename WeightType_, index_t InternalLoadUnroll_, index_t ExpertTile_ = 0>
using ck_tile::MoeSortingProblem< IndexType_, WeightType_, InternalLoadUnroll_, ExpertTile_ >::WeightType = remove_cvref_t<WeightType_>

Member Data Documentation

◆ ExpertTile

template<typename IndexType_, typename WeightType_, index_t InternalLoadUnroll_, index_t ExpertTile_ = 0>
index_t ck_tile::MoeSortingProblem< IndexType_, WeightType_, InternalLoadUnroll_, ExpertTile_ >::ExpertTile = ExpertTile_
staticconstexpr

◆ InternalLoadUnroll

template<typename IndexType_, typename WeightType_, index_t InternalLoadUnroll_, index_t ExpertTile_ = 0>
index_t ck_tile::MoeSortingProblem< IndexType_, WeightType_, InternalLoadUnroll_, ExpertTile_ >::InternalLoadUnroll
staticconstexpr
Initial value:
=
InternalLoadUnroll_

◆ WarpSize

template<typename IndexType_, typename WeightType_, index_t InternalLoadUnroll_, index_t ExpertTile_ = 0>
index_t ck_tile::MoeSortingProblem< IndexType_, WeightType_, InternalLoadUnroll_, ExpertTile_ >::WarpSize = get_warp_size()
staticconstexpr

◆ WarpsPerBlock

template<typename IndexType_, typename WeightType_, index_t InternalLoadUnroll_, index_t ExpertTile_ = 0>
index_t ck_tile::MoeSortingProblem< IndexType_, WeightType_, InternalLoadUnroll_, ExpertTile_ >::WarpsPerBlock = 1
staticconstexpr
The documentation for this struct was generated from the following file: