FmhaFwdBatchModeDropoutKargs Struct Reference#
ck_tile::FmhaFwdKernel< FmhaPipeline_, EpiloguePipeline_ >::FmhaFwdBatchModeDropoutKargs Struct Reference
#include <fmha_fwd_kernel.hpp>
Inheritance diagram for ck_tile::FmhaFwdKernel< FmhaPipeline_, EpiloguePipeline_ >::FmhaFwdBatchModeDropoutKargs:
Public Attributes | |
| ck_tile::index_t | batch_stride_randval = 0 |
| Public Attributes inherited from ck_tile::FmhaFwdKernel< FmhaPipeline_, EpiloguePipeline_ >::FmhaFwdCommonDropoutKargs | |
| float | rp_undrop = 1 |
| uint8_t | p_undrop_in_uint8_t = std::numeric_limits<uint8_t>::max() |
| bool | is_store_randval = false |
| void * | rand_val_ptr = nullptr |
| ck_tile::index_t | stride_randval = 0 |
| ck_tile::index_t | nhead_stride_randval = 0 |
| Public Attributes inherited from ck_tile::FmhaFwdKernel< FmhaPipeline_, EpiloguePipeline_ >::FmhaFwdDropoutSeedOffset | |
| ValueOrPointer< uint64_t > | drop_seed |
| ValueOrPointer< uint64_t > | drop_offset |
| bool | is_drop_seed_offset_from_host |
Additional Inherited Members | |
| Public Member Functions inherited from ck_tile::FmhaFwdKernel< FmhaPipeline_, EpiloguePipeline_ >::FmhaFwdCommonDropoutKargs | |
| void | init_dropout (float p_drop, uint64_t seed, uint64_t offset) |
| void | init_dropout (float p_drop, const uint64_t *seed_ptr, const uint64_t *offset_ptr) |
Member Data Documentation
◆ batch_stride_randval
template<typename FmhaPipeline_, typename EpiloguePipeline_>
| ck_tile::index_t ck_tile::FmhaFwdKernel< FmhaPipeline_, EpiloguePipeline_ >::FmhaFwdBatchModeDropoutKargs::batch_stride_randval = 0 |
The documentation for this struct was generated from the following file: