BlockNormReduceSync< Problem_, Policy_ > Struct Template Reference

BlockNormReduceSync&lt; Problem_, Policy_ &gt; Struct Template Reference#

Composable Kernel: ck_tile::BlockNormReduceSync< Problem_, Policy_ > Struct Template Reference
Public Types | Public Member Functions | Static Public Attributes | List of all members
ck_tile::BlockNormReduceSync< Problem_, Policy_ > Struct Template Reference

#include <block_norm_reduce.hpp>

Public Types

using Problem = remove_cvref_t<Problem_>

Public Member Functions

template<typename MeanDistributedTensor_, typename VarDistributedTensor_>
CK_TILE_DEVICE void operator() (MeanDistributedTensor_ &mean_tensor, VarDistributedTensor_ &var_tensor, int &count)

Static Public Attributes

static constexpr bool kFastFDiv = Problem::kFastFDiv
static constexpr bool kWelford = Problem::kWelford

Member Typedef Documentation

◆ Problem

template<typename Problem_, typename Policy_ = void>
using ck_tile::BlockNormReduceSync< Problem_, Policy_ >::Problem = remove_cvref_t<Problem_>

Member Function Documentation

◆ operator()()

template<typename Problem_, typename Policy_ = void>
template<typename MeanDistributedTensor_, typename VarDistributedTensor_>
CK_TILE_DEVICE void ck_tile::BlockNormReduceSync< Problem_, Policy_ >::operator() ( MeanDistributedTensor_ & mean_tensor,
VarDistributedTensor_ & var_tensor,
int & count )
inline

Member Data Documentation

◆ kFastFDiv

template<typename Problem_, typename Policy_ = void>
bool ck_tile::BlockNormReduceSync< Problem_, Policy_ >::kFastFDiv = Problem::kFastFDiv
staticconstexpr

◆ kWelford

template<typename Problem_, typename Policy_ = void>
bool ck_tile::BlockNormReduceSync< Problem_, Policy_ >::kWelford = Problem::kWelford
staticconstexpr
The documentation for this struct was generated from the following file: