internal Namespace Reference

internal Namespace Reference#

Composable Kernel: ck_tile::internal Namespace Reference
ck_tile::internal Namespace Reference

Classes

struct  ParseEnvVal
 
struct  ParseEnvVal< bool >
 
struct  ParseEnvVal< uint64_t >
 
struct  ParseEnvVal< std::string >
 
struct  EnvVar
 

Functions

template<size_t N>
bool is_any_of (const char *const (&names)[N], const std::string &str)
 
__device__ float exp2_soft_sign_impl (float softmax_scale, float logits, float logits_soft_cap_rcp)
 

Function Documentation

◆ exp2_soft_sign_impl()

__device__ float ck_tile::internal::exp2_soft_sign_impl ( float  softmax_scale,
float  logits,
float  logits_soft_cap_rcp 
)
inline

◆ is_any_of()

template<size_t N>
bool ck_tile::internal::is_any_of ( const char *const (&)  names[N],
const std::string &  str 
)