AddAddFastGelu Struct Reference

AddAddFastGelu Struct Reference#

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

#include <element_wise_operation.hpp>

Public Member Functions

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

Member Function Documentation

◆ operator()() [1/6]

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

◆ operator()() [2/6]

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

◆ operator()() [3/6]

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

◆ operator()() [4/6]

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

◆ operator()() [5/6]

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

◆ operator()() [6/6]

template<>
__host__ constexpr __device__ void ck::tensor_operation::element_wise::AddAddFastGelu::operator() ( int8_t &  e,
const int32_t &  c,
const int8_t &  d0,
const int8_t &  d1 
) const
inlineconstexpr
The documentation for this struct was generated from the following file:
  • /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck/tensor_operation/gpu/element/element_wise_operation.hpp