MultiplyFastGelu Struct Reference

MultiplyFastGelu Struct Reference#

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

#include <binary_element_wise_operation.hpp>

Public Member Functions

template<typename E, typename C, typename D>
__host__ __device__ constexpr void operator() (E &e, const C &c, const D &d) const
template<>
__host__ __device__ constexpr void operator() (float &e, const float &c, const float &d) const
template<>
__host__ __device__ constexpr void operator() (half_t &e, const half_t &c, const half_t &d) const
template<>
__host__ __device__ constexpr void operator() (half_t &e, const float &c, const half_t &d) const
template<>
__host__ __device__ constexpr void operator() (bhalf_t &e, const bhalf_t &c, const bhalf_t &d) const
template<>
__host__ __device__ constexpr void operator() (bhalf_t &e, const float &c, const bhalf_t &d) const

Static Public Attributes

static constexpr const char * name = "MultiplyFastGelu"

Member Function Documentation

◆ operator()() [1/6]

template<>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::MultiplyFastGelu::operator() ( bhalf_t & e,
const bhalf_t & c,
const bhalf_t & d ) const
inlineconstexpr

◆ operator()() [2/6]

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

◆ operator()() [3/6]

template<typename E, typename C, typename D>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::MultiplyFastGelu::operator() ( E & e,
const C & c,
const D & d ) const
constexpr

◆ operator()() [4/6]

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

◆ operator()() [5/6]

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

◆ operator()() [6/6]

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

Member Data Documentation

◆ name

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