TopkSoftmaxWarpPerRowPipeline< Problem_, Policy_ > Struct Template Reference

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

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

#include <topk_softmax_warp_per_row_pipeline.hpp>

Public Types

using Problem = remove_cvref_t<Problem_>
using Policy = remove_cvref_t<Policy_>
using WeightType = typename Problem::WeightType

Public Member Functions

template<typename InputWindow, typename OutputWindow, typename IndexWindow>
CK_TILE_DEVICE auto operator() (const InputWindow &input_window, OutputWindow &out_window, IndexWindow &idx_window, index_t rows, index_t experts, index_t k, index_t block_row_id)

Member Typedef Documentation

◆ Policy

template<typename Problem_, typename Policy_ = TopkSoftmaxWarpPerRowPolicy>
using ck_tile::TopkSoftmaxWarpPerRowPipeline< Problem_, Policy_ >::Policy = remove_cvref_t<Policy_>

◆ Problem

template<typename Problem_, typename Policy_ = TopkSoftmaxWarpPerRowPolicy>
using ck_tile::TopkSoftmaxWarpPerRowPipeline< Problem_, Policy_ >::Problem = remove_cvref_t<Problem_>

◆ WeightType

template<typename Problem_, typename Policy_ = TopkSoftmaxWarpPerRowPolicy>
using ck_tile::TopkSoftmaxWarpPerRowPipeline< Problem_, Policy_ >::WeightType = typename Problem::WeightType

Member Function Documentation

◆ operator()()

template<typename Problem_, typename Policy_ = TopkSoftmaxWarpPerRowPolicy>
template<typename InputWindow, typename OutputWindow, typename IndexWindow>
CK_TILE_DEVICE auto ck_tile::TopkSoftmaxWarpPerRowPipeline< Problem_, Policy_ >::operator() ( const InputWindow & input_window,
OutputWindow & out_window,
IndexWindow & idx_window,
index_t rows,
index_t experts,
index_t k,
index_t block_row_id )
inline
The documentation for this struct was generated from the following file: