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:
  • /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck_tile/ops/topk_softmax/pipeline/topk_softmax_warp_per_row_pipeline.hpp