BlockFmhaSplitKVCombinePipelineTileSizes< OaccDataType_, kN1_ > Struct Template Reference

BlockFmhaSplitKVCombinePipelineTileSizes&lt; OaccDataType_, kN1_ &gt; Struct Template Reference#

Composable Kernel: ck_tile::BlockFmhaSplitKVCombinePipelineTileSizes< OaccDataType_, kN1_ > Struct Template Reference
ck_tile::BlockFmhaSplitKVCombinePipelineTileSizes< OaccDataType_, kN1_ > Struct Template Reference

#include <block_fmha_pipeline_problem.hpp>

Inheritance diagram for ck_tile::BlockFmhaSplitKVCombinePipelineTileSizes< OaccDataType_, kN1_ >:
ck_tile::BlockFmhaSplitKVCombinePipelineProblem< LSEDataType_, OaccDataType_, ODataType_, HeadDimV_, kIsGroupMode_, kN1_, Traits_ >

Static Public Attributes

static constexpr index_t MaxVectorSize = 16 / sizeof(OaccDataType_)
static constexpr index_t kN1 = kN1_
static constexpr index_t NThreads = kN1 / MaxVectorSize
static constexpr index_t kM0 = get_warp_size() / NThreads

Member Data Documentation

◆ kM0

template<typename OaccDataType_, ck_tile::index_t kN1_>
index_t ck_tile::BlockFmhaSplitKVCombinePipelineTileSizes< OaccDataType_, kN1_ >::kM0 = get_warp_size() / NThreads
staticconstexpr

◆ kN1

template<typename OaccDataType_, ck_tile::index_t kN1_>
index_t ck_tile::BlockFmhaSplitKVCombinePipelineTileSizes< OaccDataType_, kN1_ >::kN1 = kN1_
staticconstexpr

◆ MaxVectorSize

template<typename OaccDataType_, ck_tile::index_t kN1_>
index_t ck_tile::BlockFmhaSplitKVCombinePipelineTileSizes< OaccDataType_, kN1_ >::MaxVectorSize = 16 / sizeof(OaccDataType_)
staticconstexpr

◆ NThreads

template<typename OaccDataType_, ck_tile::index_t kN1_>
index_t ck_tile::BlockFmhaSplitKVCombinePipelineTileSizes< OaccDataType_, kN1_ >::NThreads = kN1 / MaxVectorSize
staticconstexpr

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