/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 Source File#
topk_softmax_kernel.hpp
Go to the documentation of this file.
Definition: cluster_descriptor.hpp:13
constexpr CK_TILE_HOST_DEVICE auto pad_tensor_view(const TensorView &tensor_view, const TileLengths &tile_lengths, DoPads)
Definition: tensor_view.hpp:530
remove_cv_t< std::remove_reference_t< T > > remove_cvref_t
Definition: type_traits.hpp:21
constexpr CK_TILE_DEVICE auto make_tile_window(null_tensor_view, const WindowLengths &window_lengths, const multi_index< WindowLengths::size()> &, Ts &&...)
Definition: null_tile_window.hpp:75
constexpr CK_TILE_HOST_DEVICE auto make_tuple(Xs &&... xs)
Definition: tuple.hpp:360
Definition: topk_softmax_kernel.hpp:16
index_t stride_output
Definition: topk_softmax_kernel.hpp:24
index_t stride_input
Definition: topk_softmax_kernel.hpp:23
Definition: topk_softmax_kernel.hpp:40
const void * p_input
Definition: topk_softmax_kernel.hpp:41
void * p_output
Definition: topk_softmax_kernel.hpp:42
index_t stride_output
Definition: topk_softmax_kernel.hpp:48
index_t stride_input
Definition: topk_softmax_kernel.hpp:47
index_t num_rows
Definition: topk_softmax_kernel.hpp:44
void * p_indices
Definition: topk_softmax_kernel.hpp:43
index_t topk
Definition: topk_softmax_kernel.hpp:46
index_t num_experts
Definition: topk_softmax_kernel.hpp:45
Definition: topk_softmax_kernel.hpp:29
static constexpr index_t kBlockSize
Definition: topk_softmax_kernel.hpp:37
remove_cvref_t< typename Pipeline::Problem > Problem
Definition: topk_softmax_kernel.hpp:31
remove_cvref_t< Pipeline_ > Pipeline
Definition: topk_softmax_kernel.hpp:30
typename Problem::InputType InputType
Definition: topk_softmax_kernel.hpp:33
static constexpr CK_TILE_HOST auto MakeKargs(const Hargs &h)
Definition: topk_softmax_kernel.hpp:76
static constexpr CK_TILE_HOST auto GridSize(const Hargs &h)
Definition: topk_softmax_kernel.hpp:54
static constexpr CK_TILE_HOST_DEVICE auto BlockSize()
Definition: topk_softmax_kernel.hpp:90
typename Problem::WeightType WeightType
Definition: topk_softmax_kernel.hpp:34
CK_TILE_DEVICE void operator()(Kargs kargs) const
Definition: topk_softmax_kernel.hpp:92
typename Problem::IndexType IndexType
Definition: topk_softmax_kernel.hpp:35
Definition: integral_constant.hpp:13
Definition: sequence.hpp:49