/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck_tile/ops/smoothquant/kernel/moe_smoothquant_kernel.hpp Source File#
moe_smoothquant_kernel.hpp
Go to the documentation of this file.
122 return _SS_("moe_smoothquant_") + _SS_(t2s<XDataType>::name) + "_" + _SS_(t2s<QYDataType>::name) + "_" +
123 _TS_(S_::Block_M) + "x" + _TS_(S_::Block_N) + "_" + _TS_(S_::WarpPerBlock_M) + "x" + _TS_(S_::WarpPerBlock_N) + "_" +
124 _TS_(S_::Warp_M) + "x" + _TS_(S_::Warp_N) + "_" + _TS_(S_::Vector_M) + "x" + _TS_(S_::Vector_N) + "_" +
#define _TS_
#define _SS_
Definition: cluster_descriptor.hpp:13
constexpr CK_TILE_HOST_DEVICE auto integer_divide_ceil(X x, Y y)
Definition: math.hpp:149
constexpr CK_TILE_HOST_DEVICE auto pad_tensor_view(const TensorView &tensor_view, const TileLengths &tile_lengths, DoPads)
Definition: tensor_view.hpp:480
remove_cv_t< std::remove_reference_t< T > > remove_cvref_t
Definition: type_traits.hpp:20
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:72
constexpr CK_TILE_HOST_DEVICE auto make_tuple(Xs &&... xs)
Definition: tuple.hpp:337
Definition: moe_smoothquant_kernel.hpp:58
const void * p_smscale
Definition: moe_smoothquant_kernel.hpp:60
const void * p_topk_ids
Definition: moe_smoothquant_kernel.hpp:61
index_t hidden_size
Definition: moe_smoothquant_kernel.hpp:67
Definition: moe_smoothquant_kernel.hpp:98
Definition: moe_smoothquant_kernel.hpp:13
index_t hidden_size
Definition: moe_smoothquant_kernel.hpp:22
const void * p_topk_ids
Definition: moe_smoothquant_kernel.hpp:16
const void * p_smscale
Definition: moe_smoothquant_kernel.hpp:15
Definition: moe_smoothquant_kernel.hpp:32
static constexpr bool kTwoPass
Definition: moe_smoothquant_kernel.hpp:46
remove_cvref_t< typename Problem::SmoothScaleDataType > SmoothScaleDataType
Definition: moe_smoothquant_kernel.hpp:37
CK_TILE_DEVICE void operator()(Kargs kargs) const
Definition: moe_smoothquant_kernel.hpp:131
remove_cvref_t< typename Problem::YScaleDataType > YScaleDataType
Definition: moe_smoothquant_kernel.hpp:39
static constexpr CK_TILE_HOST Kargs MakeKargs(const Hargs &hargs)
Definition: moe_smoothquant_kernel.hpp:75
remove_cvref_t< typename Problem::QYDataType > QYDataType
Definition: moe_smoothquant_kernel.hpp:40
remove_cvref_t< typename Problem::ComputeDataType > ComputeDataType
Definition: moe_smoothquant_kernel.hpp:38
remove_cvref_t< Pipeline_ > Pipeline
Definition: moe_smoothquant_kernel.hpp:33
remove_cvref_t< typename Problem::XDataType > XDataType
Definition: moe_smoothquant_kernel.hpp:36
static constexpr index_t Vector_N
Definition: moe_smoothquant_kernel.hpp:49
static constexpr CK_TILE_HOST auto BlockSize()
Definition: moe_smoothquant_kernel.hpp:95
static constexpr index_t Block_N
Definition: moe_smoothquant_kernel.hpp:43
static CK_TILE_HOST std::string GetName()
Definition: moe_smoothquant_kernel.hpp:110
typename Pipeline::Problem Problem
Definition: moe_smoothquant_kernel.hpp:34
static constexpr CK_TILE_HOST auto GridSize(const Hargs &hargs)
Definition: moe_smoothquant_kernel.hpp:90
static constexpr index_t Repeat_N
Definition: moe_smoothquant_kernel.hpp:50
static constexpr index_t ThreadPerWarp_N
Definition: moe_smoothquant_kernel.hpp:48
static constexpr CK_TILE_HOST_DEVICE index_t GetSmemSize()
Definition: moe_smoothquant_kernel.hpp:108
static constexpr index_t Block_M
Definition: moe_smoothquant_kernel.hpp:42
Definition: integral_constant.hpp:13
Definition: sequence.hpp:52