EpilogueGraph< Steps, EpilogueTypes > Struct Template Reference

• ck_tile
• EpilogueGraph

Public Member Functions | Public Attributes | List of all members

ck_tile::EpilogueGraph< Steps, EpilogueTypes > Struct Template Reference

Operation graph that sequentially composes multiple epilogue operations. More...

#include <epilogue_chainer.hpp>

Public Member Functions
constexpr	EpilogueGraph ()=default
constexpr	EpilogueGraph (EpilogueTypes... eps)
template<typename OutWindow , typename AccTile , typename AuxWindows , typename Context >
CK_TILE_DEVICE void	operator() (OutWindow &out_window, const AccTile &acc_tile, const AuxWindows &aux_windows, void *p_smem, Context &context) const
	Process all epilogues for each iteration in sequence. More...

Public Attributes
ck_tile::tuple< EpilogueTypes... >	epilogues

Detailed Description

template<index_t Steps, typename... EpilogueTypes>
struct ck_tile::EpilogueGraph< Steps, EpilogueTypes >

Operation graph that sequentially composes multiple epilogue operations.

Template Parameters

Steps	Number of iterations
EpilogueTypes	Types of epilogue nodes in the operation graph

Constructor & Destructor Documentation

◆ EpilogueGraph() [1/2]

template<index_t Steps, typename... EpilogueTypes>

constexpr ck_tile::EpilogueGraph< Steps, EpilogueTypes >::EpilogueGraph ( )

constexprdefault

◆ EpilogueGraph() [2/2]

template<index_t Steps, typename... EpilogueTypes>

constexpr ck_tile::EpilogueGraph< Steps, EpilogueTypes >::EpilogueGraph ( EpilogueTypes...  eps )

inlineconstexpr

Member Function Documentation

◆ operator()()

template<index_t Steps, typename... EpilogueTypes>

template<typename OutWindow , typename AccTile , typename AuxWindows , typename Context >

CK_TILE_DEVICE void ck_tile::EpilogueGraph< Steps, EpilogueTypes >::operator() ( OutWindow &  out_window, const AccTile &  acc_tile, const AuxWindows &  aux_windows, void *  p_smem, Context &  context  ) const

inline

Process all epilogues for each iteration in sequence.

Member Data Documentation

◆ epilogues

template<index_t Steps, typename... EpilogueTypes>

ck_tile::tuple<EpilogueTypes...> ck_tile::EpilogueGraph< Steps, EpilogueTypes >::epilogues

---

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/epilogue/chainer/epilogue_chainer.hpp