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

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

Composable Kernel: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/epilogue.hpp Source File
epilogue.hpp
Go to the documentation of this file.
1 // Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
2 // SPDX-License-Identifier: MIT
3 #pragma once
4 
5 #include "ck_tile/ops/epilogue/chainer/common_epilogue_ops.hpp"
6 #include "ck_tile/ops/epilogue/chainer/cshuffle_epilogue_chainer_ops.hpp"
7 #include "ck_tile/ops/epilogue/chainer/cshuffle_epilogue_schedule.hpp"
8 #include "ck_tile/ops/epilogue/chainer/epilogue_chainer.hpp"
9 #include "ck_tile/ops/epilogue/cshuffle_epilogue.hpp"
10 #include "ck_tile/ops/epilogue/default_2d_and_dynamic_quant_epilogue.hpp"
11 #include "ck_tile/ops/epilogue/default_2d_epilogue.hpp"
12 #include "ck_tile/ops/epilogue/dynamic_quant_epilogue.hpp"
13 #include "ck_tile/ops/common/generic_2d_block_shape.hpp"
14 #include "ck_tile/ops/common/load_interleaved_pk_type.hpp"
15 #include "ck_tile/ops/common/streamk_common.hpp"
16 #include "ck_tile/ops/common/tensor_layout.hpp"
17 #include "ck_tile/ops/common/utils.hpp"
common_epilogue_ops.hpp
Reusable simple epilogue operations which might be used to compose more complex one.
cshuffle_epilogue.hpp
cshuffle_epilogue_chainer_ops.hpp
cshuffle_epilogue_schedule.hpp
default_2d_and_dynamic_quant_epilogue.hpp
default_2d_epilogue.hpp
dynamic_quant_epilogue.hpp
epilogue_chainer.hpp
generic_2d_block_shape.hpp
load_interleaved_pk_type.hpp
streamk_common.hpp
tensor_layout.hpp