#include <block_dropout.hpp>
|
CK_TILE_HOST_DEVICE | BlockDropout (index_t i_batch, index_t i_head, index_t nheads, unsigned long long seed, unsigned long long offset, float rp_undrop_, uint8_t p_undrop_in_uint8_t_, bool is_store_randval_) |
|
template<typename BlockGemm , typename PComputeDataType , typename RandValOutputDataType , typename PComputeWindow , typename RandValDramWindow > |
CK_TILE_HOST_DEVICE void | Run (void *randval_ptr, const index_t start_n0_idx, PComputeWindow &p_compute, RandValDramWindow &randval_dram_window) const |
|
◆ BlockDropout()
CK_TILE_HOST_DEVICE ck_tile::BlockDropout::BlockDropout |
( |
index_t |
i_batch, |
|
|
index_t |
i_head, |
|
|
index_t |
nheads, |
|
|
unsigned long long |
seed, |
|
|
unsigned long long |
offset, |
|
|
float |
rp_undrop_, |
|
|
uint8_t |
p_undrop_in_uint8_t_, |
|
|
bool |
is_store_randval_ |
|
) |
| |
|
inline |
◆ MakeRandvalDramWindow()
template<typename BlockGemm , bool IsFwd = true, typename RandValDramBlockWindowTmp >
static constexpr CK_TILE_HOST_DEVICE auto ck_tile::BlockDropout::MakeRandvalDramWindow |
( |
RandValDramBlockWindowTmp & |
randval_dram_block_window_tmp, |
|
|
index_t |
seqlen_qk_start |
|
) |
| |
|
inlinestaticconstexpr |
◆ MakeRandValLdsBlockDescriptor()
template<typename BlockGemm >
static constexpr CK_TILE_HOST_DEVICE auto ck_tile::BlockDropout::MakeRandValLdsBlockDescriptor |
( |
| ) |
|
|
inlinestaticconstexpr |
◆ MakeRandValLdsShuffleTileDistribution()
template<typename BlockGemm >
static constexpr CK_TILE_HOST_DEVICE auto ck_tile::BlockDropout::MakeRandValLdsShuffleTileDistribution |
( |
| ) |
|
|
inlinestaticconstexpr |
◆ MakeRandValTileDistribution()
template<typename BlockGemm >
static constexpr CK_TILE_HOST_DEVICE auto ck_tile::BlockDropout::MakeRandValTileDistribution |
( |
| ) |
|
|
inlinestaticconstexpr |
◆ Run()
template<typename BlockGemm , typename PComputeDataType , typename RandValOutputDataType , typename PComputeWindow , typename RandValDramWindow >
CK_TILE_HOST_DEVICE void ck_tile::BlockDropout::Run |
( |
void * |
randval_ptr, |
|
|
const index_t |
start_n0_idx, |
|
|
PComputeWindow & |
p_compute, |
|
|
RandValDramWindow & |
randval_dram_window |
|
) |
| const |
|
inline |
◆ is_store_randval
const bool ck_tile::BlockDropout::is_store_randval |
◆ p_undrop_in_uint8_t
const uint8_t ck_tile::BlockDropout::p_undrop_in_uint8_t |
◆ ph
◆ rp_undrop
const float ck_tile::BlockDropout::rp_undrop |
The documentation for this struct was generated from the following file:
- /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck_tile/ops/fmha/block/block_dropout.hpp