ScaleScalarOp Struct Reference

ScaleScalarOp Struct Reference#

Composable Kernel: ck_tile::ScaleScalarOp Struct Reference
ck_tile::ScaleScalarOp Struct Reference

Scale working tile by scalar values. More...

#include <common_epilogue_ops.hpp>

Public Member Functions

template<typename OutWindow , typename AccTile , typename AuxWindows , typename IAccess , typename Context , typename ScaleA , typename ScaleB >
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, const ScaleA &scale_a, const ScaleB &scale_b)
 

Detailed Description

Scale working tile by scalar values.

Context Requirements
working_tile: Tile to scale

Member Function Documentation

◆ operator()()

template<typename OutWindow , typename AccTile , typename AuxWindows , typename IAccess , typename Context , typename ScaleA , typename ScaleB >
CK_TILE_DEVICE void ck_tile::ScaleScalarOp::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,
const ScaleA &  scale_a,
const ScaleB &  scale_b 
)
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