RoPEKargs Struct Reference

RoPEKargs Struct Reference#

Composable Kernel: ck_tile::FmhaFwdAppendKVKernel< FmhaPipeline_ >::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: