FastGelu Struct Reference

FastGelu Struct Reference#

Composable Kernel: ck_tile::element_wise::FastGelu Struct Reference
Public Member Functions | Static Public Attributes | List of all members
ck_tile::element_wise::FastGelu Struct Reference

#include <unary_element_wise_operation.hpp>

Public Member Functions

template<typename Y, typename X>
CK_TILE_HOST void operator() (Y &y, const X &x) const
template<typename Y, typename X>
CK_TILE_DEVICE void operator() (Y &y, const X &x) const
template<>
CK_TILE_HOST void operator() (float &y, const float &x) const
template<>
CK_TILE_DEVICE void operator() (float &y, const float &x) const
template<>
CK_TILE_HOST void operator() (ck_tile::fp16_t &y, const ck_tile::fp16_t &x) const
template<>
CK_TILE_DEVICE void operator() (ck_tile::fp16_t &y, const ck_tile::fp16_t &x) const
template<>
CK_TILE_HOST void operator() (ck_tile::fp16_t &y, const float &x) const
template<>
CK_TILE_DEVICE void operator() (ck_tile::fp16_t &y, const float &x) const
template<>
CK_TILE_HOST void operator() (ck_tile::bf16_t &y, const float &x) const
template<>
CK_TILE_DEVICE void operator() (ck_tile::bf16_t &y, const float &x) const
template<>
CK_TILE_DEVICE void operator() (ck_tile::bf16_t &y, const ck_tile::bf16_t &x) const
template<>
CK_TILE_HOST void operator() (ck_tile::bf16_t &y, const ck_tile::bf16_t &x) const

Static Public Attributes

static constexpr const char * name = "FastGelu"

Member Function Documentation

◆ operator()() [1/12]

template<>
CK_TILE_HOST void ck_tile::element_wise::FastGelu::operator() ( ck_tile::bf16_t & y,
const ck_tile::bf16_t & x ) const
inline

◆ operator()() [2/12]

template<>
CK_TILE_DEVICE void ck_tile::element_wise::FastGelu::operator() ( ck_tile::bf16_t & y,
const ck_tile::bf16_t & x ) const
inline

◆ operator()() [3/12]

template<>
CK_TILE_DEVICE void ck_tile::element_wise::FastGelu::operator() ( ck_tile::bf16_t & y,
const float & x ) const
inline

◆ operator()() [4/12]

template<>
CK_TILE_HOST void ck_tile::element_wise::FastGelu::operator() ( ck_tile::bf16_t & y,
const float & x ) const
inline

◆ operator()() [5/12]

template<>
CK_TILE_DEVICE void ck_tile::element_wise::FastGelu::operator() ( ck_tile::fp16_t & y,
const ck_tile::fp16_t & x ) const
inline

◆ operator()() [6/12]

template<>
CK_TILE_HOST void ck_tile::element_wise::FastGelu::operator() ( ck_tile::fp16_t & y,
const ck_tile::fp16_t & x ) const
inline

◆ operator()() [7/12]

template<>
CK_TILE_DEVICE void ck_tile::element_wise::FastGelu::operator() ( ck_tile::fp16_t & y,
const float & x ) const
inline

◆ operator()() [8/12]

template<>
CK_TILE_HOST void ck_tile::element_wise::FastGelu::operator() ( ck_tile::fp16_t & y,
const float & x ) const
inline

◆ operator()() [9/12]

template<>
CK_TILE_DEVICE void ck_tile::element_wise::FastGelu::operator() ( float & y,
const float & x ) const
inline

◆ operator()() [10/12]

template<>
CK_TILE_HOST void ck_tile::element_wise::FastGelu::operator() ( float & y,
const float & x ) const
inline

◆ operator()() [11/12]

template<typename Y, typename X>
CK_TILE_DEVICE void ck_tile::element_wise::FastGelu::operator() ( Y & y,
const X & x ) const

◆ operator()() [12/12]

template<typename Y, typename X>
CK_TILE_HOST void ck_tile::element_wise::FastGelu::operator() ( Y & y,
const X & x ) const

Member Data Documentation

◆ name

const char* ck_tile::element_wise::FastGelu::name = "FastGelu"
staticconstexpr
The documentation for this struct was generated from the following file: