gpu::xetla::subgroup::rng_dropout_op_t< dtype_mask, arch_tag, enable > Struct Template Reference

Is the random number generator and dropout op functor. More...

#include <tile_op_functor.hpp>

Detailed Description

template<typename dtype_mask, gpu_arch arch_tag, class enable = void>
struct gpu::xetla::subgroup::rng_dropout_op_t< dtype_mask, arch_tag, enable >

Is the random number generator and dropout op functor.

Generate the mask data and get input from matAcc, do the scaling and zero out, update the output in place, dump the mask buffer to memory. Used in epilogue::tile_op or chained_tile_op.

Template Parameters

dtype_mask	Is the mask data type.
arch_tag	Is the hardware architecture tag.