gpu::xetla::subgroup::gelu_fwd_op_t Struct Reference

Is the element-wise gelu inference forward op functor. More...

#include <tile_op_functor.hpp>

Classes
struct	arguments_t

Public Member Functions
template<typename matAcc_t , typename coord_t >
__XETLA_API KERNEL_FUNC void	operator() (matAcc_t &matAcc, const coord_t &coord, const arguments_t &args, uint32_t slm_base=0, uint32_t nbarrier_base=0)

Detailed Description

Is the element-wise gelu inference forward op functor.

Get the gelu input from matAcc, update the the gelu output in place, Used in epilogue::tile_op or chained_tile_op.

Member Function Documentation

operator()()

template<typename matAcc_t , typename coord_t >

__XETLA_API KERNEL_FUNC void gpu::xetla::subgroup::gelu_fwd_op_t::operator() ( matAcc_t &  matAcc, const coord_t &  coord, const arguments_t &  args, uint32_t  slm_base = 0, uint32_t  nbarrier_base = 0  )

inline