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: