Layernorm2dFwdPipelineDefaultPolicy Struct Reference

Layernorm2dFwdPipelineDefaultPolicy Struct Reference#

Composable Kernel: ck_tile::Layernorm2dFwdPipelineDefaultPolicy Struct Reference
Static Public Member Functions | List of all members
ck_tile::Layernorm2dFwdPipelineDefaultPolicy Struct Reference

#include <layernorm2d_fwd_pipeline_default_policy.hpp>

Static Public Member Functions

template<typename Problem>
static CK_TILE_DEVICE constexpr auto MakeXBlockTileDistribution ()
template<typename Problem>
static CK_TILE_DEVICE constexpr auto MakeGammaBetaBlockTileDistribution ()
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr auto GetBlockNormReduce ()
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr auto GetBlockNormReduceSync ()
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr auto GetBlockNormReduceCrossWarpSync ()
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr index_t GetSmemSize ()

Member Function Documentation

◆ GetBlockNormReduce()

template<typename Problem>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::Layernorm2dFwdPipelineDefaultPolicy::GetBlockNormReduce ( )
inlinestaticconstexpr

◆ GetBlockNormReduceCrossWarpSync()

template<typename Problem>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::Layernorm2dFwdPipelineDefaultPolicy::GetBlockNormReduceCrossWarpSync ( )
inlinestaticconstexpr

◆ GetBlockNormReduceSync()

template<typename Problem>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::Layernorm2dFwdPipelineDefaultPolicy::GetBlockNormReduceSync ( )
inlinestaticconstexpr

◆ GetSmemSize()

template<typename Problem>
CK_TILE_HOST_DEVICE constexpr index_t ck_tile::Layernorm2dFwdPipelineDefaultPolicy::GetSmemSize ( )
inlinestaticconstexpr

◆ MakeGammaBetaBlockTileDistribution()

template<typename Problem>
CK_TILE_DEVICE constexpr auto ck_tile::Layernorm2dFwdPipelineDefaultPolicy::MakeGammaBetaBlockTileDistribution ( )
inlinestaticconstexpr

◆ MakeXBlockTileDistribution()

template<typename Problem>
CK_TILE_DEVICE constexpr auto ck_tile::Layernorm2dFwdPipelineDefaultPolicy::MakeXBlockTileDistribution ( )
inlinestaticconstexpr
The documentation for this struct was generated from the following file: