gru_config_t Class Reference

#include <kernel_func.hpp>

Public Types
using	dtype_in = bf16
using	dtype_acc = float

Static Public Attributes
static constexpr uint32_t	layer_size = 3
	layer_size = 3
static constexpr uint32_t	sequence_length = 2
	sequence_length = 64
static constexpr uint32_t	batch_size = 512
	batch_size = 512
static constexpr uint32_t	input_size = 384
	input_size = 384
static constexpr uint32_t	hidden_size = 704
	hidden_size = 688;
static constexpr uint32_t	wg_tile_m = 64
	launch config
static constexpr uint32_t	wg_tile_n = 128
static constexpr uint32_t	sg_tile_m = 16
static constexpr uint32_t	sg_tile_n = 16
static constexpr uint32_t	sg_tile_k = 32

Member Typedef Documentation

dtype_acc

using gru_config_t::dtype_acc = float

dtype_in

using gru_config_t::dtype_in = bf16

Member Data Documentation

batch_size

constexpr uint32_t gru_config_t::batch_size = 512

staticconstexpr

batch_size = 512

hidden_size

constexpr uint32_t gru_config_t::hidden_size = 704

staticconstexpr

hidden_size = 688;

input_size

constexpr uint32_t gru_config_t::input_size = 384

staticconstexpr

input_size = 384

layer_size

constexpr uint32_t gru_config_t::layer_size = 3

staticconstexpr

layer_size = 3

sequence_length

constexpr uint32_t gru_config_t::sequence_length = 2

staticconstexpr

sequence_length = 64

sg_tile_k

constexpr uint32_t gru_config_t::sg_tile_k = 32

staticconstexpr

sg_tile_m

constexpr uint32_t gru_config_t::sg_tile_m = 16

staticconstexpr

sg_tile_n

constexpr uint32_t gru_config_t::sg_tile_n = 16

staticconstexpr

wg_tile_m

constexpr uint32_t gru_config_t::wg_tile_m = 64

staticconstexpr

launch config

wg_tile_n

constexpr uint32_t gru_config_t::wg_tile_n = 128

staticconstexpr