TopkSoftmaxKernel< Pipeline_ > Struct Template Reference

TopkSoftmaxKernel&lt; Pipeline_ &gt; Struct Template Reference#

Composable Kernel: ck_tile::TopkSoftmaxKernel< Pipeline_ > Struct Template Reference
Classes | Public Types | Public Member Functions | Static Public Member Functions | Static Public Attributes | List of all members
ck_tile::TopkSoftmaxKernel< Pipeline_ > Struct Template Reference

#include <topk_softmax_kernel.hpp>

Classes

struct  TopkSoftmaxKargs
 

Public Types

using Pipeline = remove_cvref_t< Pipeline_ >
 
using Problem = remove_cvref_t< typename Pipeline::Problem >
 
using InputType = typename Problem::InputType
 
using WeightType = typename Problem::WeightType
 
using IndexType = typename Problem::IndexType
 
using Kargs = TopkSoftmaxKargs
 
using Hargs = TopkSoftmaxHostArgs
 

Public Member Functions

CK_TILE_DEVICE void operator() (Kargs kargs) const
 

Static Public Member Functions

static constexpr CK_TILE_HOST auto GridSize (const Hargs &h)
 
static constexpr CK_TILE_HOST auto MakeKargs (const Hargs &h)
 
static constexpr CK_TILE_HOST_DEVICE auto BlockSize ()
 

Static Public Attributes

static constexpr index_t kBlockSize = Problem::BlockSize
 

Member Typedef Documentation

◆ Hargs

template<typename Pipeline_ >
using ck_tile::TopkSoftmaxKernel< Pipeline_ >::Hargs = TopkSoftmaxHostArgs

◆ IndexType

template<typename Pipeline_ >
using ck_tile::TopkSoftmaxKernel< Pipeline_ >::IndexType = typename Problem::IndexType

◆ InputType

template<typename Pipeline_ >
using ck_tile::TopkSoftmaxKernel< Pipeline_ >::InputType = typename Problem::InputType

◆ Kargs

template<typename Pipeline_ >
using ck_tile::TopkSoftmaxKernel< Pipeline_ >::Kargs = TopkSoftmaxKargs

◆ Pipeline

template<typename Pipeline_ >
using ck_tile::TopkSoftmaxKernel< Pipeline_ >::Pipeline = remove_cvref_t<Pipeline_>

◆ Problem

template<typename Pipeline_ >
using ck_tile::TopkSoftmaxKernel< Pipeline_ >::Problem = remove_cvref_t<typename Pipeline::Problem>

◆ WeightType

template<typename Pipeline_ >
using ck_tile::TopkSoftmaxKernel< Pipeline_ >::WeightType = typename Problem::WeightType

Member Function Documentation

◆ BlockSize()

template<typename Pipeline_ >
static constexpr CK_TILE_HOST_DEVICE auto ck_tile::TopkSoftmaxKernel< Pipeline_ >::BlockSize ( )
inlinestaticconstexpr

◆ GridSize()

template<typename Pipeline_ >
static constexpr CK_TILE_HOST auto ck_tile::TopkSoftmaxKernel< Pipeline_ >::GridSize ( const Hargs &  h)
inlinestaticconstexpr

◆ MakeKargs()

template<typename Pipeline_ >
static constexpr CK_TILE_HOST auto ck_tile::TopkSoftmaxKernel< Pipeline_ >::MakeKargs ( const Hargs &  h)
inlinestaticconstexpr

◆ operator()()

template<typename Pipeline_ >
CK_TILE_DEVICE void ck_tile::TopkSoftmaxKernel< Pipeline_ >::operator() ( Kargs  kargs) const
inline

Member Data Documentation

◆ kBlockSize

template<typename Pipeline_ >
constexpr index_t ck_tile::TopkSoftmaxKernel< Pipeline_ >::kBlockSize = Problem::BlockSize
staticconstexpr
The documentation for this struct was generated from the following file:
  • /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/topk_softmax/kernel/topk_softmax_kernel.hpp