HotLoopScheduler< Problem > Class Template Reference

HotLoopScheduler&lt; Problem &gt; Class Template Reference#

Composable Kernel: ck_tile::BlockFmhaBwdPipelineTrLoadDefaultPolicy::HotLoopScheduler< Problem > Class Template Reference
Static Public Member Functions | Static Public Attributes | List of all members
ck_tile::BlockFmhaBwdPipelineTrLoadDefaultPolicy::HotLoopScheduler< Problem > Class Template Reference

#include <block_fmha_bwd_pipeline_trload_default_policy.hpp>

Static Public Member Functions

static CK_TILE_DEVICE constexpr void SchedulerGemm0 ()
static CK_TILE_DEVICE constexpr void SchedulerGemm12 ()
static CK_TILE_DEVICE constexpr void SchedulerGemm3 ()
static CK_TILE_DEVICE constexpr void SchedulerGemm4 ()

Static Public Attributes

static constexpr index_t TOTAL_VMEM_READ

Member Function Documentation

◆ SchedulerGemm0()

template<typename Problem>
CK_TILE_DEVICE constexpr void ck_tile::BlockFmhaBwdPipelineTrLoadDefaultPolicy::HotLoopScheduler< Problem >::SchedulerGemm0 ( )
inlinestaticconstexpr

◆ SchedulerGemm12()

template<typename Problem>
CK_TILE_DEVICE constexpr void ck_tile::BlockFmhaBwdPipelineTrLoadDefaultPolicy::HotLoopScheduler< Problem >::SchedulerGemm12 ( )
inlinestaticconstexpr

◆ SchedulerGemm3()

template<typename Problem>
CK_TILE_DEVICE constexpr void ck_tile::BlockFmhaBwdPipelineTrLoadDefaultPolicy::HotLoopScheduler< Problem >::SchedulerGemm3 ( )
inlinestaticconstexpr

◆ SchedulerGemm4()

template<typename Problem>
CK_TILE_DEVICE constexpr void ck_tile::BlockFmhaBwdPipelineTrLoadDefaultPolicy::HotLoopScheduler< Problem >::SchedulerGemm4 ( )
inlinestaticconstexpr

Member Data Documentation

◆ TOTAL_VMEM_READ

template<typename Problem>
index_t ck_tile::BlockFmhaBwdPipelineTrLoadDefaultPolicy::HotLoopScheduler< Problem >::TOTAL_VMEM_READ
staticconstexpr
Initial value:
=
Q_VMEM_READ + OGrad_VMEM_READ + LSE_VMEM_READ + D_VMEM_READ + DQ_VMEM_WRITE
The documentation for this class was generated from the following file: