BlockDropoutBwd< false, IsWG32_, IsStoreRandval_ > Struct Template Reference

BlockDropoutBwd&lt; false, IsWG32_, IsStoreRandval_ &gt; Struct Template Reference#

Composable Kernel: ck_tile::BlockDropoutBwd< false, IsWG32_, IsStoreRandval_ > Struct Template Reference
Static Public Member Functions | Static Public Attributes | List of all members
ck_tile::BlockDropoutBwd< false, IsWG32_, IsStoreRandval_ > Struct Template Reference

#include <block_dropout.hpp>

Static Public Member Functions

template<typename BlockGemm, bool IsFwd = false, typename RandValDramBlockWindowTmp>
static CK_TILE_HOST_DEVICE constexpr auto MakeRandvalDramWindow (RandValDramBlockWindowTmp &randval_dram_block_window_tmp, index_t seqlen_qk_start)

Static Public Attributes

static constexpr bool IsDropout = false
static constexpr bool IsStoreRandval = IsStoreRandval_

Member Function Documentation

◆ MakeRandvalDramWindow()

template<bool IsWG32_, bool IsStoreRandval_>
template<typename BlockGemm, bool IsFwd = false, typename RandValDramBlockWindowTmp>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::BlockDropoutBwd< false, IsWG32_, IsStoreRandval_ >::MakeRandvalDramWindow ( RandValDramBlockWindowTmp & randval_dram_block_window_tmp,
index_t seqlen_qk_start )
inlinestaticconstexpr

Member Data Documentation

◆ IsDropout

template<bool IsWG32_, bool IsStoreRandval_>
bool ck_tile::BlockDropoutBwd< false, IsWG32_, IsStoreRandval_ >::IsDropout = false
staticconstexpr

◆ IsStoreRandval

template<bool IsWG32_, bool IsStoreRandval_>
bool ck_tile::BlockDropoutBwd< false, IsWG32_, IsStoreRandval_ >::IsStoreRandval = IsStoreRandval_
staticconstexpr
The documentation for this struct was generated from the following file: