/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/epilogue/chainer/common_epilogue_ops.hpp File Reference

• develop
• include
• ck_tile
• ops
• epilogue
• chainer

Classes | Namespaces

common_epilogue_ops.hpp File Reference

Reusable simple epilogue operations which might be used to compose more complex one. More...

#include "ck_tile/core.hpp"
#include "ck_tile/ops/common/tensor_layout.hpp"

Go to the source code of this file.

Classes
struct	ck_tile::ScaleScalarOp
	Scale working tile by scalar values. More...
struct	ck_tile::CastAndStoreToLdsOp< DataType >
	Cast working tile and store to LDS. More...
struct	ck_tile::LoadFromLdsOp< TileEncodingPattern >
	Load output tile from LDS with synchronization. More...
struct	ck_tile::ElementwiseOp< Elementwise, NumAux >
	Apply elementwise operation with auxiliary tensors. More...
struct	ck_tile::StoreOp< MemOp >
	Store output tile to global memory. More...
struct	ck_tile::MoveWindowsOp< SFC, NumAux >
	Move output and auxiliary windows by step from space-filling curve. More...

Namespaces
	ck_tile

Detailed Description

Reusable simple epilogue operations which might be used to compose more complex one.

Context Interface

Operations expect the context to provide:

• working_tile: Tile for intermediate computations
• out_tile: Output tile for final results
• aux_windows: Tuple of auxiliary tensor windows (e.g., D tensors)
• lds_write_window: Window for writing to LDS (if using LDS)
• lds_read_window: Window for reading from LDS (if using LDS)