gpu::xetla::group::mask_gen_t< dtype_mask_, wg_tile_n_, wg_tile_m_, sg_tile_n_, sg_tile_m_, random_simd_, arch_ >::arguments_t Struct Reference

#include <dropout_mask_gen.hpp>

Public Attributes
dtype_mask *	mask_ptr
uint32_t	matrix_m
uint32_t	matrix_n
uint32_t	mask_ld
uint64_t	rand_seed = 67280421310721
uint64_t *	rand_offset_ptr
float	dropout_prob

Member Data Documentation

dropout_prob

template<typename dtype_mask_ , uint32_t wg_tile_n_, uint32_t wg_tile_m_, uint32_t sg_tile_n_, uint32_t sg_tile_m_, uint32_t random_simd_ = 16, gpu_arch arch_ = gpu_arch::Xe>

float gpu::xetla::group::mask_gen_t< dtype_mask_, wg_tile_n_, wg_tile_m_, sg_tile_n_, sg_tile_m_, random_simd_, arch_ >::arguments_t::dropout_prob

mask_ld

template<typename dtype_mask_ , uint32_t wg_tile_n_, uint32_t wg_tile_m_, uint32_t sg_tile_n_, uint32_t sg_tile_m_, uint32_t random_simd_ = 16, gpu_arch arch_ = gpu_arch::Xe>

uint32_t gpu::xetla::group::mask_gen_t< dtype_mask_, wg_tile_n_, wg_tile_m_, sg_tile_n_, sg_tile_m_, random_simd_, arch_ >::arguments_t::mask_ld

mask_ptr

template<typename dtype_mask_ , uint32_t wg_tile_n_, uint32_t wg_tile_m_, uint32_t sg_tile_n_, uint32_t sg_tile_m_, uint32_t random_simd_ = 16, gpu_arch arch_ = gpu_arch::Xe>

dtype_mask* gpu::xetla::group::mask_gen_t< dtype_mask_, wg_tile_n_, wg_tile_m_, sg_tile_n_, sg_tile_m_, random_simd_, arch_ >::arguments_t::mask_ptr

matrix_m

template<typename dtype_mask_ , uint32_t wg_tile_n_, uint32_t wg_tile_m_, uint32_t sg_tile_n_, uint32_t sg_tile_m_, uint32_t random_simd_ = 16, gpu_arch arch_ = gpu_arch::Xe>

uint32_t gpu::xetla::group::mask_gen_t< dtype_mask_, wg_tile_n_, wg_tile_m_, sg_tile_n_, sg_tile_m_, random_simd_, arch_ >::arguments_t::matrix_m

matrix_n

template<typename dtype_mask_ , uint32_t wg_tile_n_, uint32_t wg_tile_m_, uint32_t sg_tile_n_, uint32_t sg_tile_m_, uint32_t random_simd_ = 16, gpu_arch arch_ = gpu_arch::Xe>

uint32_t gpu::xetla::group::mask_gen_t< dtype_mask_, wg_tile_n_, wg_tile_m_, sg_tile_n_, sg_tile_m_, random_simd_, arch_ >::arguments_t::matrix_n

rand_offset_ptr

template<typename dtype_mask_ , uint32_t wg_tile_n_, uint32_t wg_tile_m_, uint32_t sg_tile_n_, uint32_t sg_tile_m_, uint32_t random_simd_ = 16, gpu_arch arch_ = gpu_arch::Xe>

uint64_t* gpu::xetla::group::mask_gen_t< dtype_mask_, wg_tile_n_, wg_tile_m_, sg_tile_n_, sg_tile_m_, random_simd_, arch_ >::arguments_t::rand_offset_ptr

rand_seed

template<typename dtype_mask_ , uint32_t wg_tile_n_, uint32_t wg_tile_m_, uint32_t sg_tile_n_, uint32_t sg_tile_m_, uint32_t random_simd_ = 16, gpu_arch arch_ = gpu_arch::Xe>

uint64_t gpu::xetla::group::mask_gen_t< dtype_mask_, wg_tile_n_, wg_tile_m_, sg_tile_n_, sg_tile_m_, random_simd_, arch_ >::arguments_t::rand_seed = 67280421310721