BlockTopkStream2D< Problem_, Policy_ > Struct Template Reference

BlockTopkStream2D< Problem_, Policy_ > Struct Template Reference#

Composable Kernel: ck_tile::BlockTopkStream2D< Problem_, Policy_ > Struct Template Reference

#include <block_topk_stream_2d.hpp>

Classes
struct	ArgmaxPacket

Public Types
using	Problem = remove_cvref_t<Problem_>
using	Policy = remove_cvref_t<Policy_>
using	DataType = typename Problem::DataType
using	IndexType = typename Problem::IndexType

Public Member Functions
template<typename DistributedTensor, typename OutWindow, typename IdxWindow, index_t dim = 1>
CK_TILE_DEVICE void	operator() (const DistributedTensor &x, const OutWindow &out_window, const IdxWindow &idx_window, index_t k, number< dim >={})

Member Typedef Documentation

◆ DataType

template<typename Problem_, typename Policy_ = void>

using ck_tile::BlockTopkStream2D< Problem_, Policy_ >::DataType = typename Problem::DataType

◆ IndexType

template<typename Problem_, typename Policy_ = void>

using ck_tile::BlockTopkStream2D< Problem_, Policy_ >::IndexType = typename Problem::IndexType

◆ Policy

template<typename Problem_, typename Policy_ = void>

using ck_tile::BlockTopkStream2D< Problem_, Policy_ >::Policy = remove_cvref_t<Policy_>

◆ Problem

template<typename Problem_, typename Policy_ = void>

using ck_tile::BlockTopkStream2D< Problem_, Policy_ >::Problem = remove_cvref_t<Problem_>

Member Function Documentation

◆ operator()()

template<typename Problem_, typename Policy_ = void>

template<typename DistributedTensor, typename OutWindow, typename IdxWindow, index_t dim = 1>

CK_TILE_DEVICE void ck_tile::BlockTopkStream2D< Problem_, Policy_ >::operator()	(	const DistributedTensor &	x,
		const OutWindow &	out_window,
		const IdxWindow &	idx_window,
		index_t	k,
		number< dim >	= {} )

inline

The documentation for this struct was generated from the following file:

block_topk_stream_2d.hpp