block_norm_reduce.hpp File Reference

• include
• ck_tile
• ops
• norm_reduce
• block

Classes | Namespaces | Functions

block_norm_reduce.hpp File Reference

#include "ck_tile/core.hpp"
#include "ck_tile/ops/norm_reduce/thread/thread_welford.hpp"

Go to the source code of this file.

Classes
struct	ck_tile::BlockNormReduce< Problem_, Policy_ >
struct	ck_tile::BlockNormReduceSync< Problem_, Policy_ >
struct	ck_tile::BlockNormReduceCrossWarpSync< Problem_, Policy_ >

Namespaces
namespace	ck_tile

Functions
template<typename BlockShape>
CK_TILE_DEVICE constexpr index_t	ck_tile::block_tile_welford_calculate_max_count (int row_size)
template<typename VarDistributedTensor_, bool FastFdiv_ = false>
CK_TILE_DEVICE constexpr void	ck_tile::block_tile_welford_post_scale_var (VarDistributedTensor_ &var_tensor, int count, bool_constant< FastFdiv_ >={})