layer_norm_fwd_xe.hpp File Reference

C++ API. More...

#include "experimental/group/fused_op/layer_norm_fused_op_fwd_xe.hpp"
#include "experimental/kernel/layer_norm/api.hpp"
#include "experimental/kernel/layer_norm/common.hpp"
#include "experimental/kernel/layer_norm/config.hpp"

Include dependency graph for layer_norm_fwd_xe.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	gpu::xetla::kernel::layer_norm_fwd_t< dtype_x_, dtype_y_, dtype_weight_, dtype_acc_, layer_norm_attr_, store_for_bwd_, gpu_arch::Xe, ln_fwd_fused_op_ >
struct	gpu::xetla::kernel::layer_norm_fwd_t< dtype_x_, dtype_y_, dtype_weight_, dtype_acc_, layer_norm_attr_, store_for_bwd_, gpu_arch::Xe, ln_fwd_fused_op_ >::get_barrier_count
struct	gpu::xetla::kernel::layer_norm_fwd_t< dtype_x_, dtype_y_, dtype_weight_, dtype_acc_, layer_norm_attr_, store_for_bwd_, gpu_arch::Xe, ln_fwd_fused_op_ >::get_slm_size
struct	gpu::xetla::kernel::layer_norm_fwd_t< dtype_x_, dtype_y_, dtype_weight_, dtype_acc_, layer_norm_attr_, store_for_bwd_, gpu_arch::Xe, ln_fwd_fused_op_ >::arguments_t
struct	gpu::xetla::kernel::layer_norm_fwd_t< dtype_x_, dtype_y_, dtype_weight_, dtype_acc_, layer_norm_attr_, store_for_bwd_, gpu_arch::Xe, ln_fwd_fused_op_ >::parallel_mu_m2_t< T, SZ, N >
struct	gpu::xetla::kernel::layer_norm_fwd_t< dtype_x_, dtype_y_, dtype_weight_, dtype_acc_, layer_norm_attr_, store_for_bwd_, gpu_arch::Xe, ln_fwd_fused_op_ >::parallel_mu_m2_t< T, 1, N >

Namespaces
namespace	gpu
namespace	gpu::xetla
namespace	gpu::xetla::kernel

Detailed Description

C++ API.