StandardAttentionParams< ImplMask > Struct Template Reference

StandardAttentionParams&lt; ImplMask &gt; Struct Template Reference#

Composable Kernel: ck_tile::StandardAttentionParams< ImplMask > Struct Template Reference
ck_tile::StandardAttentionParams< ImplMask > Struct Template Reference

#include <variants.hpp>

Public Member Functions

__device__ __host__ StandardAttentionParams (const ImplMask &impl_mask_, float sm_scale_)
 

Public Attributes

const ImplMask & impl_mask
 
float sm_scale
 

Constructor & Destructor Documentation

◆ StandardAttentionParams()

template<typename ImplMask >
__device__ __host__ ck_tile::StandardAttentionParams< ImplMask >::StandardAttentionParams ( const ImplMask &  impl_mask_,
float  sm_scale_ 
)
inline

Member Data Documentation

◆ impl_mask

template<typename ImplMask >
const ImplMask& ck_tile::StandardAttentionParams< ImplMask >::impl_mask

◆ sm_scale

template<typename ImplMask >
float ck_tile::StandardAttentionParams< ImplMask >::sm_scale

The documentation for this struct was generated from the following file:
  • /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/fmha/block/variants.hpp