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 >
|
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