LogitsSoftCapKargs Struct Reference

• ck_tile
• FmhaFwdSplitKVKernel
• LogitsSoftCapKargs

Public Member Functions | Public Attributes | List of all members

ck_tile::FmhaFwdSplitKVKernel< FmhaPipeline_, EpiloguePipeline_ >::LogitsSoftCapKargs Struct Reference

#include <fmha_fwd_splitkv_kernel.hpp>

Public Member Functions
	LogitsSoftCapKargs ()=default
void	init_logits_soft_cap (float logits_soft_cap_)

Public Attributes
float	logits_soft_cap
float	logits_soft_cap_rcp

Constructor & Destructor Documentation

◆ LogitsSoftCapKargs()

template<typename FmhaPipeline_, typename EpiloguePipeline_>

ck_tile::FmhaFwdSplitKVKernel< FmhaPipeline_, EpiloguePipeline_ >::LogitsSoftCapKargs::LogitsSoftCapKargs ( )

default

Member Function Documentation

◆ init_logits_soft_cap()

template<typename FmhaPipeline_, typename EpiloguePipeline_>

void ck_tile::FmhaFwdSplitKVKernel< FmhaPipeline_, EpiloguePipeline_ >::LogitsSoftCapKargs::init_logits_soft_cap ( float logits_soft_cap_ )

inline

Member Data Documentation

◆ logits_soft_cap

template<typename FmhaPipeline_, typename EpiloguePipeline_>

float ck_tile::FmhaFwdSplitKVKernel< FmhaPipeline_, EpiloguePipeline_ >::LogitsSoftCapKargs::logits_soft_cap

◆ logits_soft_cap_rcp

template<typename FmhaPipeline_, typename EpiloguePipeline_>

float ck_tile::FmhaFwdSplitKVKernel< FmhaPipeline_, EpiloguePipeline_ >::LogitsSoftCapKargs::logits_soft_cap_rcp

---

The documentation for this struct was generated from the following file:

• fmha_fwd_splitkv_kernel.hpp