MultiplyAddFastGelu Struct Reference

MultiplyAddFastGelu Struct Reference#

Composable Kernel: ck::tensor_operation::element_wise::MultiplyAddFastGelu Struct Reference
Public Member Functions | Static Public Attributes | List of all members
ck::tensor_operation::element_wise::MultiplyAddFastGelu Struct Reference

#include <element_wise_operation.hpp>

Public Member Functions

template<typename E, typename C, typename D0, typename D1>
__host__ __device__ constexpr void operator() (E &e, const C &c, const D0 &d0, const D1 &d1) const
template<>
__host__ __device__ constexpr void operator() (ck::bhalf_t &e, const float &c, const ck::bhalf_t &d0, const ck::bhalf_t &d1) const

Static Public Attributes

static constexpr const char * name = "MultiplyAddFastGelu"

Member Function Documentation

◆ operator()() [1/2]

template<>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::MultiplyAddFastGelu::operator() ( ck::bhalf_t & e,
const float & c,
const ck::bhalf_t & d0,
const ck::bhalf_t & d1 ) const
inlineconstexpr

◆ operator()() [2/2]

template<typename E, typename C, typename D0, typename D1>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::MultiplyAddFastGelu::operator() ( E & e,
const C & c,
const D0 & d0,
const D1 & d1 ) const
constexpr

Member Data Documentation

◆ name

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