BlockReduce2dSync< Problem_, Policy_ > Struct Template Reference

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

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

#include <block_reduce2d.hpp>

Public Types

using Problem = remove_cvref_t<Problem_>

Public Member Functions

template<typename YDistributedTensor_, typename ReduceFunc>
CK_TILE_DEVICE void operator() (YDistributedTensor_ &y_tensor, const ReduceFunc &reduce_func)
template<typename YDistributedTensor_, typename YIndexDistributedTensor_, typename ReduceFunc>
CK_TILE_DEVICE void operator() (YDistributedTensor_ &y_tensor, YIndexDistributedTensor_ &y_index_tensor, const ReduceFunc &reduce_func)

Member Typedef Documentation

◆ Problem

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

Member Function Documentation

◆ operator()() [1/2]

template<typename Problem_, typename Policy_ = void>
template<typename YDistributedTensor_, typename ReduceFunc>
CK_TILE_DEVICE void ck_tile::BlockReduce2dSync< Problem_, Policy_ >::operator() ( YDistributedTensor_ & y_tensor,
const ReduceFunc & reduce_func )
inline

◆ operator()() [2/2]

template<typename Problem_, typename Policy_ = void>
template<typename YDistributedTensor_, typename YIndexDistributedTensor_, typename ReduceFunc>
CK_TILE_DEVICE void ck_tile::BlockReduce2dSync< Problem_, Policy_ >::operator() ( YDistributedTensor_ & y_tensor,
YIndexDistributedTensor_ & y_index_tensor,
const ReduceFunc & reduce_func )
inline
The documentation for this struct was generated from the following file: