FusedMoeGemmTilePartitioner_Linear< BlockShape_ > Struct Template Reference

• ck_tile
• FusedMoeGemmTilePartitioner_Linear

Public Types | Public Member Functions | Static Public Member Functions | Static Public Attributes | List of all members

ck_tile::FusedMoeGemmTilePartitioner_Linear< BlockShape_ > Struct Template Reference

#include <fused_moegemm_tile_partitioner.hpp>

Public Types
using	BlockShape = ck_tile::remove_cvref_t<BlockShape_>

Public Member Functions
CK_TILE_DEVICE auto	operator() (ck_tile::index_t, ck_tile::index_t)

Static Public Member Functions
static CK_TILE_HOST constexpr auto	GridSize (index_t max_tokens, index_t intermediate_size)

Static Public Attributes
static constexpr const char *	name = "lin"

Member Typedef Documentation

◆ BlockShape

template<typename BlockShape_>

using ck_tile::FusedMoeGemmTilePartitioner_Linear< BlockShape_ >::BlockShape = ck_tile::remove_cvref_t<BlockShape_>

Member Function Documentation

◆ GridSize()

template<typename BlockShape_>

CK_TILE_HOST constexpr auto ck_tile::FusedMoeGemmTilePartitioner_Linear< BlockShape_ >::GridSize ( index_t max_tokens, index_t intermediate_size )

inlinestaticconstexpr

◆ operator()()

template<typename BlockShape_>

CK_TILE_DEVICE auto ck_tile::FusedMoeGemmTilePartitioner_Linear< BlockShape_ >::operator() ( ck_tile::index_t , ck_tile::index_t  )

inline

Member Data Documentation

◆ name

template<typename BlockShape_>

const char* ck_tile::FusedMoeGemmTilePartitioner_Linear< BlockShape_ >::name = "lin"

staticconstexpr

---

The documentation for this struct was generated from the following file:

• fused_moegemm_tile_partitioner.hpp