Default2DAndDynamicQuantEpilogue< Problem_, Policy_ > Struct Template Reference

Default2DAndDynamicQuantEpilogue&lt; Problem_, Policy_ &gt; Struct Template Reference#

Composable Kernel: ck_tile::Default2DAndDynamicQuantEpilogue< Problem_, Policy_ > Struct Template Reference
Public Types | Public Member Functions | Static Public Member Functions | Static Public Attributes | List of all members
ck_tile::Default2DAndDynamicQuantEpilogue< Problem_, Policy_ > Struct Template Reference

#include <default_2d_and_dynamic_quant_epilogue.hpp>

Public Types

using Problem = remove_cvref_t<Problem_>
using AccDataType = remove_cvref_t<typename Problem::AccDataType>
using UnquantYDataType = remove_cvref_t<typename Problem::UnquantYDataType>
using Default2DProblem
using Default2D = Default2DEpilogue<Default2DProblem>
using DynamicQuant = DynamicQuantEpilogue<Problem>

Public Member Functions

template<typename ODramWindowTmpD, typename ODramWindowTmpQ, typename SmoothScaleWindow, typename YScaleWindow, typename OAccTile>
CK_TILE_DEVICE auto operator() (ODramWindowTmpD &o_direct_dram_window_tmp, ODramWindowTmpQ &o_quant_dram_window_tmp, const SmoothScaleWindow &sm_scale_window_, YScaleWindow &y_scale_window, const OAccTile &o_acc_tile, void *smem)
template<typename ODramWindowTmpD, typename ODramWindowTmpQ, typename YScaleWindow, typename OAccTile>
CK_TILE_DEVICE auto operator() (ODramWindowTmpD &o_direct_dram_window_tmp, ODramWindowTmpQ &o_quant_dram_window_tmp, YScaleWindow &y_scale_window, const OAccTile &o_acc_tile, void *smem)

Static Public Member Functions

static CK_TILE_HOST_DEVICE constexpr index_t GetSmemSize ()

Static Public Attributes

static constexpr bool kPadM = Problem::Traits::kPadM
static constexpr bool kPadN = Problem::Traits::kPadN
static constexpr bool UseRawStore = Problem::Traits::UseRawStore

Member Typedef Documentation

◆ AccDataType

template<typename Problem_, typename Policy_ = void>
using ck_tile::Default2DAndDynamicQuantEpilogue< Problem_, Policy_ >::AccDataType = remove_cvref_t<typename Problem::AccDataType>

◆ Default2D

template<typename Problem_, typename Policy_ = void>
using ck_tile::Default2DAndDynamicQuantEpilogue< Problem_, Policy_ >::Default2D = Default2DEpilogue<Default2DProblem>

◆ Default2DProblem

template<typename Problem_, typename Policy_ = void>
using ck_tile::Default2DAndDynamicQuantEpilogue< Problem_, Policy_ >::Default2DProblem
Initial value:
Default2DEpilogueProblem<AccDataType, UnquantYDataType, kPadM, kPadN, UseRawStore>
ck_tile::Default2DEpilogueProblem
Definition default_2d_epilogue.hpp:21

◆ DynamicQuant

template<typename Problem_, typename Policy_ = void>
using ck_tile::Default2DAndDynamicQuantEpilogue< Problem_, Policy_ >::DynamicQuant = DynamicQuantEpilogue<Problem>

◆ Problem

template<typename Problem_, typename Policy_ = void>
using ck_tile::Default2DAndDynamicQuantEpilogue< Problem_, Policy_ >::Problem = remove_cvref_t<Problem_>

◆ UnquantYDataType

template<typename Problem_, typename Policy_ = void>
using ck_tile::Default2DAndDynamicQuantEpilogue< Problem_, Policy_ >::UnquantYDataType = remove_cvref_t<typename Problem::UnquantYDataType>

Member Function Documentation

◆ GetSmemSize()

template<typename Problem_, typename Policy_ = void>
CK_TILE_HOST_DEVICE constexpr index_t ck_tile::Default2DAndDynamicQuantEpilogue< Problem_, Policy_ >::GetSmemSize ( )
inlinestaticconstexpr

◆ operator()() [1/2]

template<typename Problem_, typename Policy_ = void>
template<typename ODramWindowTmpD, typename ODramWindowTmpQ, typename SmoothScaleWindow, typename YScaleWindow, typename OAccTile>
CK_TILE_DEVICE auto ck_tile::Default2DAndDynamicQuantEpilogue< Problem_, Policy_ >::operator() ( ODramWindowTmpD & o_direct_dram_window_tmp,
ODramWindowTmpQ & o_quant_dram_window_tmp,
const SmoothScaleWindow & sm_scale_window_,
YScaleWindow & y_scale_window,
const OAccTile & o_acc_tile,
void * smem )
inline

◆ operator()() [2/2]

template<typename Problem_, typename Policy_ = void>
template<typename ODramWindowTmpD, typename ODramWindowTmpQ, typename YScaleWindow, typename OAccTile>
CK_TILE_DEVICE auto ck_tile::Default2DAndDynamicQuantEpilogue< Problem_, Policy_ >::operator() ( ODramWindowTmpD & o_direct_dram_window_tmp,
ODramWindowTmpQ & o_quant_dram_window_tmp,
YScaleWindow & y_scale_window,
const OAccTile & o_acc_tile,
void * smem )
inline

Member Data Documentation

◆ kPadM

template<typename Problem_, typename Policy_ = void>
bool ck_tile::Default2DAndDynamicQuantEpilogue< Problem_, Policy_ >::kPadM = Problem::Traits::kPadM
staticconstexpr

◆ kPadN

template<typename Problem_, typename Policy_ = void>
bool ck_tile::Default2DAndDynamicQuantEpilogue< Problem_, Policy_ >::kPadN = Problem::Traits::kPadN
staticconstexpr

◆ UseRawStore

template<typename Problem_, typename Policy_ = void>
bool ck_tile::Default2DAndDynamicQuantEpilogue< Problem_, Policy_ >::UseRawStore = Problem::Traits::UseRawStore
staticconstexpr
The documentation for this struct was generated from the following file: