include/ck_tile/ops/rmsnorm2d.hpp Source File

• include
• ck_tile
• ops

rmsnorm2d.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/rmsnorm2d/kernel/rmsnorm2d_fwd_kernel.hpp"

7 #include "ck_tile/ops/rmsnorm2d/pipeline/rmsnorm2d_fwd_pipeline_default_policy.hpp"

8 #include "ck_tile/ops/rmsnorm2d/pipeline/rmsnorm2d_fwd_pipeline_one_pass.hpp"

9 #include "ck_tile/ops/rmsnorm2d/pipeline/rmsnorm2d_fwd_pipeline_problem.hpp"

10 #include "ck_tile/ops/rmsnorm2d/pipeline/rmsnorm2d_fwd_pipeline_two_pass.hpp"

11 #include "ck_tile/ops/rmsnorm2d/pipeline/rmsnorm2d_fwd_traits.hpp"

12 #include "ck_tile/ops/common/generic_2d_block_shape.hpp"

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

generic_2d_block_shape.hpp

rmsnorm2d_fwd_kernel.hpp

rmsnorm2d_fwd_pipeline_default_policy.hpp

rmsnorm2d_fwd_pipeline_one_pass.hpp

rmsnorm2d_fwd_pipeline_problem.hpp

rmsnorm2d_fwd_pipeline_two_pass.hpp

rmsnorm2d_fwd_traits.hpp

tensor_layout.hpp