LogitsSoftCapParams< ImplMask, UseExp2 > Struct Template Reference

LogitsSoftCapParams< ImplMask, UseExp2 > Struct Template Reference#

Composable Kernel: ck_tile::LogitsSoftCapParams< ImplMask, UseExp2 > Struct Template Reference

#include <variants.hpp>

Public Member Functions
__device__	LogitsSoftCapParams (const ImplMask &impl_mask_, float sm_scale_, float logits_soft_cap_)
__host__	LogitsSoftCapParams (const ImplMask &impl_mask_, float sm_scale_, float logits_soft_cap_)
__device__ __host__	LogitsSoftCapParams (const ImplMask &impl_mask_, float sm_scale_, float logits_soft_cap_, float logits_soft_cap_rcp_)

Public Attributes
const ImplMask &	impl_mask
float	sm_scale
float	logits_soft_cap
float	logits_soft_cap_rcp

Constructor & Destructor Documentation

◆ LogitsSoftCapParams() [1/3]

template<typename ImplMask, bool UseExp2 = false>

__device__ ck_tile::LogitsSoftCapParams< ImplMask, UseExp2 >::LogitsSoftCapParams	(	const ImplMask &	impl_mask_,
		float	sm_scale_,
		float	logits_soft_cap_ )

inline

◆ LogitsSoftCapParams() [2/3]

template<typename ImplMask, bool UseExp2 = false>

__host__ ck_tile::LogitsSoftCapParams< ImplMask, UseExp2 >::LogitsSoftCapParams	(	const ImplMask &	impl_mask_,
		float	sm_scale_,
		float	logits_soft_cap_ )

inline

◆ LogitsSoftCapParams() [3/3]

template<typename ImplMask, bool UseExp2 = false>

__device__ __host__ ck_tile::LogitsSoftCapParams< ImplMask, UseExp2 >::LogitsSoftCapParams	(	const ImplMask &	impl_mask_,
		float	sm_scale_,
		float	logits_soft_cap_,
		float	logits_soft_cap_rcp_ )

inline

Member Data Documentation

◆ impl_mask

template<typename ImplMask, bool UseExp2 = false>

const ImplMask& ck_tile::LogitsSoftCapParams< ImplMask, UseExp2 >::impl_mask

◆ logits_soft_cap

template<typename ImplMask, bool UseExp2 = false>

float ck_tile::LogitsSoftCapParams< ImplMask, UseExp2 >::logits_soft_cap

◆ logits_soft_cap_rcp

template<typename ImplMask, bool UseExp2 = false>

float ck_tile::LogitsSoftCapParams< ImplMask, UseExp2 >::logits_soft_cap_rcp

◆ sm_scale

template<typename ImplMask, bool UseExp2 = false>

float ck_tile::LogitsSoftCapParams< ImplMask, UseExp2 >::sm_scale

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

variants.hpp