gpu::xetla::group::ln_fwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc > Struct Template Reference

#include <layer_norm_fused_op_fwd_xe.hpp>

Public Attributes
dtype_in *	bias_ptr
dtype_in *	res_add_ptr
dtype_out *	bias_dropout_res_ptr
uint8_t *	mask_ptr
uint32_t	matrix_m
uint32_t	matrix_n
uint32_t	mat_ld
uint32_t	mask_ld
uint64_t	rand_seed = 67280421310721
uint64_t *	rand_offset_ptr
float	dropout_prob
float	dropout_scale

Detailed Description

template<typename dtype_in, typename dtype_out, typename dtype_acc>
struct gpu::xetla::group::ln_fwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >

Template Parameters

dtype_in
dtype_out
dtype_acc

Member Data Documentation

bias_dropout_res_ptr

template<typename dtype_in , typename dtype_out , typename dtype_acc >

dtype_out* gpu::xetla::group::ln_fwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::bias_dropout_res_ptr

bias_ptr

template<typename dtype_in , typename dtype_out , typename dtype_acc >

dtype_in* gpu::xetla::group::ln_fwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::bias_ptr

dropout_prob

template<typename dtype_in , typename dtype_out , typename dtype_acc >

float gpu::xetla::group::ln_fwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::dropout_prob

dropout_scale

template<typename dtype_in , typename dtype_out , typename dtype_acc >

float gpu::xetla::group::ln_fwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::dropout_scale

mask_ld

template<typename dtype_in , typename dtype_out , typename dtype_acc >

uint32_t gpu::xetla::group::ln_fwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::mask_ld

mask_ptr

template<typename dtype_in , typename dtype_out , typename dtype_acc >

uint8_t* gpu::xetla::group::ln_fwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::mask_ptr

mat_ld

template<typename dtype_in , typename dtype_out , typename dtype_acc >

uint32_t gpu::xetla::group::ln_fwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::mat_ld

matrix_m

template<typename dtype_in , typename dtype_out , typename dtype_acc >

uint32_t gpu::xetla::group::ln_fwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::matrix_m

matrix_n

template<typename dtype_in , typename dtype_out , typename dtype_acc >

uint32_t gpu::xetla::group::ln_fwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::matrix_n

rand_offset_ptr

template<typename dtype_in , typename dtype_out , typename dtype_acc >

uint64_t* gpu::xetla::group::ln_fwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::rand_offset_ptr

rand_seed

template<typename dtype_in , typename dtype_out , typename dtype_acc >

uint64_t gpu::xetla::group::ln_fwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::rand_seed = 67280421310721

res_add_ptr

template<typename dtype_in , typename dtype_out , typename dtype_acc >

dtype_in* gpu::xetla::group::ln_fwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::res_add_ptr