layernorm2d.hpp Source File# Composable Kernel: layernorm2d.hpp Source File includeck_tileops layernorm2d.hpp Go to the documentation of this file. 1// SPDX-License-Identifier: MIT 2// Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved. 3 4#pragma once 5 6#include "ck_tile/ops/layernorm2d/kernel/layernorm2d_fwd_kernel.hpp" 7#include "ck_tile/ops/layernorm2d/pipeline/layernorm2d_fwd_pipeline_default_policy.hpp" 8#include "ck_tile/ops/layernorm2d/pipeline/layernorm2d_fwd_pipeline_one_pass.hpp" 9#include "ck_tile/ops/layernorm2d/pipeline/layernorm2d_fwd_pipeline_problem.hpp" 10#include "ck_tile/ops/layernorm2d/pipeline/layernorm2d_fwd_pipeline_two_pass.hpp" 11#include "ck_tile/ops/layernorm2d/pipeline/layernorm2d_fwd_traits.hpp" 12#include "ck_tile/ops/common/generic_2d_block_shape.hpp" 13#include "ck_tile/ops/common/load_interleaved_pk_type.hpp" 14#include "ck_tile/ops/common/streamk_common.hpp" 15#include "ck_tile/ops/common/tensor_layout.hpp" 16#include "ck_tile/ops/common/utils.hpp" generic_2d_block_shape.hpp layernorm2d_fwd_kernel.hpp layernorm2d_fwd_pipeline_default_policy.hpp layernorm2d_fwd_pipeline_one_pass.hpp layernorm2d_fwd_pipeline_problem.hpp layernorm2d_fwd_pipeline_two_pass.hpp layernorm2d_fwd_traits.hpp load_interleaved_pk_type.hpp streamk_common.hpp tensor_layout.hpp utils.hpp