RoPEKargs Struct Reference#
ck_tile::FmhaFwdAppendKVKernel< FmhaPipeline_ >::RoPEKargs Struct Reference
#include <fmha_fwd_appendkv_kernel.hpp>
Public Attributes | |
| const void * | rotary_cos_ptr |
| const void * | rotary_sin_ptr |
| ck_tile::index_t | rotary_dim |
| bool | has_mask |
Member Data Documentation
◆ has_mask
template<typename FmhaPipeline_>
| bool ck_tile::FmhaFwdAppendKVKernel< FmhaPipeline_ >::RoPEKargs::has_mask |
◆ rotary_cos_ptr
template<typename FmhaPipeline_>
| const void* ck_tile::FmhaFwdAppendKVKernel< FmhaPipeline_ >::RoPEKargs::rotary_cos_ptr |
◆ rotary_dim
template<typename FmhaPipeline_>
| ck_tile::index_t ck_tile::FmhaFwdAppendKVKernel< FmhaPipeline_ >::RoPEKargs::rotary_dim |
◆ rotary_sin_ptr
template<typename FmhaPipeline_>
| const void* ck_tile::FmhaFwdAppendKVKernel< FmhaPipeline_ >::RoPEKargs::rotary_sin_ptr |
The documentation for this struct was generated from the following file: