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 CK_TILE_HOST constexpr auto GridSize (const Hargs &h)
static CK_TILE_HOST constexpr auto MakeKargs (const Hargs &h)
static CK_TILE_HOST_DEVICE constexpr 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_>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::TopkSoftmaxKernel< Pipeline_ >::BlockSize ( )
inlinestaticconstexpr

◆ GridSize()

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

◆ MakeKargs()

template<typename Pipeline_>
CK_TILE_HOST constexpr 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_>
index_t ck_tile::TopkSoftmaxKernel< Pipeline_ >::kBlockSize = Problem::BlockSize
staticconstexpr
The documentation for this struct was generated from the following file: