CommonBiasKargs Struct Reference

CommonBiasKargs Struct Reference#

Composable Kernel: ck_tile::FmhaFwdSplitKVKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonBiasKargs Struct Reference
ck_tile::FmhaFwdSplitKVKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonBiasKargs Struct Reference

#include <fmha_fwd_splitkv_kernel.hpp>

Inheritance diagram for ck_tile::FmhaFwdSplitKVKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonBiasKargs:
ck_tile::FmhaFwdSplitKVKernel< FmhaPipeline_, EpiloguePipeline_ >::BatchModeBiasKargs

Public Attributes

const void * bias_ptr = nullptr
 
ck_tile::index_t stride_bias = 0
 
ck_tile::index_t nhead_stride_bias = 0
 

Member Data Documentation

◆ bias_ptr

template<typename FmhaPipeline_ , typename EpiloguePipeline_ >
const void* ck_tile::FmhaFwdSplitKVKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonBiasKargs::bias_ptr = nullptr

◆ nhead_stride_bias

template<typename FmhaPipeline_ , typename EpiloguePipeline_ >
ck_tile::index_t ck_tile::FmhaFwdSplitKVKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonBiasKargs::nhead_stride_bias = 0

◆ stride_bias

template<typename FmhaPipeline_ , typename EpiloguePipeline_ >
ck_tile::index_t ck_tile::FmhaFwdSplitKVKernel< FmhaPipeline_, EpiloguePipeline_ >::CommonBiasKargs::stride_bias = 0

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