FmhaFwdDropoutSeedOffset Struct Reference

FmhaFwdDropoutSeedOffset Struct Reference#

Composable Kernel: ck_tile::FmhaFwdKernel< FmhaPipeline_, EpiloguePipeline_ >::FmhaFwdDropoutSeedOffset Struct Reference
Classes | Public Attributes | List of all members
ck_tile::FmhaFwdKernel< FmhaPipeline_, EpiloguePipeline_ >::FmhaFwdDropoutSeedOffset Struct Reference

#include <fmha_fwd_kernel.hpp>

Inheritance diagram for ck_tile::FmhaFwdKernel< FmhaPipeline_, EpiloguePipeline_ >::FmhaFwdDropoutSeedOffset:
ck_tile::FmhaFwdKernel< FmhaPipeline_, EpiloguePipeline_ >::FmhaFwdCommonDropoutKargs ck_tile::FmhaFwdKernel< FmhaPipeline_, EpiloguePipeline_ >::FmhaFwdBatchModeDropoutKargs

Classes

union  ValueOrPointer

Public Attributes

ValueOrPointer< uint64_t > drop_seed
ValueOrPointer< uint64_t > drop_offset
bool is_drop_seed_offset_from_host

Member Data Documentation

◆ drop_offset

template<typename FmhaPipeline_, typename EpiloguePipeline_>
ValueOrPointer<uint64_t> ck_tile::FmhaFwdKernel< FmhaPipeline_, EpiloguePipeline_ >::FmhaFwdDropoutSeedOffset::drop_offset

◆ drop_seed

template<typename FmhaPipeline_, typename EpiloguePipeline_>
ValueOrPointer<uint64_t> ck_tile::FmhaFwdKernel< FmhaPipeline_, EpiloguePipeline_ >::FmhaFwdDropoutSeedOffset::drop_seed

◆ is_drop_seed_offset_from_host

template<typename FmhaPipeline_, typename EpiloguePipeline_>
bool ck_tile::FmhaFwdKernel< FmhaPipeline_, EpiloguePipeline_ >::FmhaFwdDropoutSeedOffset::is_drop_seed_offset_from_host
The documentation for this struct was generated from the following file: