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

#include <row_reduction_fused_op_xe.hpp>

Public Attributes
dtype_in *	gelu_bwd_w_ptr
dtype_out *	gelu_bwd_x_ptr
dtype_out *	dropout_bwd_ptr
uint8_t *	mask_ptr
float	dropout_prob
float	dropout_scale_inv
uint32_t	matrix_m
uint32_t	matrix_n
uint32_t	mat_in_ld
uint32_t	mat_out_ld

Detailed Description

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

Template Parameters

dtype_in
dtype_out
dtype_acc

Member Data Documentation

dropout_bwd_ptr

template<typename dtype_in , typename dtype_out , typename dtype_acc >

dtype_out* gpu::xetla::group::xetla_row_reduction_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::dropout_bwd_ptr

dropout_prob

template<typename dtype_in , typename dtype_out , typename dtype_acc >

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

dropout_scale_inv

template<typename dtype_in , typename dtype_out , typename dtype_acc >

float gpu::xetla::group::xetla_row_reduction_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::dropout_scale_inv

gelu_bwd_w_ptr

template<typename dtype_in , typename dtype_out , typename dtype_acc >

dtype_in* gpu::xetla::group::xetla_row_reduction_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::gelu_bwd_w_ptr

gelu_bwd_x_ptr

template<typename dtype_in , typename dtype_out , typename dtype_acc >

dtype_out* gpu::xetla::group::xetla_row_reduction_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::gelu_bwd_x_ptr

mask_ptr

template<typename dtype_in , typename dtype_out , typename dtype_acc >

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

mat_in_ld

template<typename dtype_in , typename dtype_out , typename dtype_acc >

uint32_t gpu::xetla::group::xetla_row_reduction_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::mat_in_ld

mat_out_ld

template<typename dtype_in , typename dtype_out , typename dtype_acc >

uint32_t gpu::xetla::group::xetla_row_reduction_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::mat_out_ld

matrix_m

template<typename dtype_in , typename dtype_out , typename dtype_acc >

uint32_t gpu::xetla::group::xetla_row_reduction_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::xetla_row_reduction_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::matrix_n