ElementwiseOp< Elementwise, NumAux > Struct Template Reference

• ck_tile
• ElementwiseOp

Public Member Functions | List of all members

ck_tile::ElementwiseOp< Elementwise, NumAux > Struct Template Reference

Apply elementwise operation with auxiliary tensors. More...

#include <common_epilogue_ops.hpp>

Public Member Functions
template<typename OutWindow , typename AccTile , typename AuxWindows , typename IAccess , typename Context >
CK_TILE_DEVICE void	operator() ([[maybe_unused]] OutWindow &out_window, [[maybe_unused]] const AccTile &acc_tile, [[maybe_unused]] const AuxWindows &aux_windows, [[maybe_unused]] void *p_smem, [[maybe_unused]] IAccess iAccess, Context &context)

Detailed Description

template<typename Elementwise, index_t NumAux>
struct ck_tile::ElementwiseOp< Elementwise, NumAux >

Apply elementwise operation with auxiliary tensors.

Template Parameters

Elementwise	Elementwise functor type
NumAux	Number of auxiliary tensors to load and apply

Context Requirements

out_tile: In/out tile for elementwise operation aux_windows: Tuple of auxiliary tensor windows

Member Function Documentation

◆ operator()()

template<typename Elementwise , index_t NumAux>

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

CK_TILE_DEVICE void ck_tile::ElementwiseOp< Elementwise, NumAux >::operator() ( [[maybe_unused] ] OutWindow &  out_window, [[maybe_unused] ] const AccTile &  acc_tile, [[maybe_unused] ] const AuxWindows &  aux_windows, [[maybe_unused] ] void *  p_smem, [[maybe_unused] ] IAccess  iAccess, Context &  context  )

inline

---

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/common_epilogue_ops.hpp