SplitKBatchOffset Struct Reference

SplitKBatchOffset Struct Reference#

Composable Kernel: ck_tile::MoeFlatmmKernel< TilePartitioner_, FlatmmPipeline_, EpiloguePipeline_, kind, FusedActivation >::SplitKBatchOffset Struct Reference
Public Member Functions | Public Attributes | List of all members
ck_tile::MoeFlatmmKernel< TilePartitioner_, FlatmmPipeline_, EpiloguePipeline_, kind, FusedActivation >::SplitKBatchOffset Struct Reference

#include <moe_flatmm_kernel.hpp>

Public Member Functions

template<class KernelArgs>
__device__ SplitKBatchOffset (const KernelArgs &kargs, const std::size_t k_id=blockIdx.z)

Public Attributes

index_t a_k_split_offset
index_t b_k_split_offset
index_t splitted_k

Constructor & Destructor Documentation

◆ SplitKBatchOffset()

template<typename TilePartitioner_, typename FlatmmPipeline_, typename EpiloguePipeline_, MoeFlatmmKind kind, typename FusedActivation = moe::MoeSilu>
template<class KernelArgs>
__device__ ck_tile::MoeFlatmmKernel< TilePartitioner_, FlatmmPipeline_, EpiloguePipeline_, kind, FusedActivation >::SplitKBatchOffset::SplitKBatchOffset ( const KernelArgs & kargs,
const std::size_t k_id = blockIdx.z )
inline

Member Data Documentation

◆ a_k_split_offset

template<typename TilePartitioner_, typename FlatmmPipeline_, typename EpiloguePipeline_, MoeFlatmmKind kind, typename FusedActivation = moe::MoeSilu>
index_t ck_tile::MoeFlatmmKernel< TilePartitioner_, FlatmmPipeline_, EpiloguePipeline_, kind, FusedActivation >::SplitKBatchOffset::a_k_split_offset

◆ b_k_split_offset

template<typename TilePartitioner_, typename FlatmmPipeline_, typename EpiloguePipeline_, MoeFlatmmKind kind, typename FusedActivation = moe::MoeSilu>
index_t ck_tile::MoeFlatmmKernel< TilePartitioner_, FlatmmPipeline_, EpiloguePipeline_, kind, FusedActivation >::SplitKBatchOffset::b_k_split_offset

◆ splitted_k

template<typename TilePartitioner_, typename FlatmmPipeline_, typename EpiloguePipeline_, MoeFlatmmKind kind, typename FusedActivation = moe::MoeSilu>
index_t ck_tile::MoeFlatmmKernel< TilePartitioner_, FlatmmPipeline_, EpiloguePipeline_, kind, FusedActivation >::SplitKBatchOffset::splitted_k
The documentation for this struct was generated from the following file: