include/ck_tile/ops/add_rmsnorm2d_rdquant.hpp Source File

• include
• ck_tile
• ops

add_rmsnorm2d_rdquant.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/add_rmsnorm2d_rdquant/kernel/add_rmsnorm2d_rdquant_fwd_kernel.hpp"

7 #include "ck_tile/ops/add_rmsnorm2d_rdquant/pipeline/add_rmsnorm2d_rdquant_fwd_pipeline_default_policy.hpp"

8 #include "ck_tile/ops/add_rmsnorm2d_rdquant/pipeline/add_rmsnorm2d_rdquant_fwd_pipeline_one_pass.hpp"

9 #include "ck_tile/ops/add_rmsnorm2d_rdquant/pipeline/add_rmsnorm2d_rdquant_fwd_pipeline_problem.hpp"

10 #include "ck_tile/ops/add_rmsnorm2d_rdquant/pipeline/add_rmsnorm2d_rdquant_fwd_pipeline_three_pass.hpp"

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

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

add_rmsnorm2d_rdquant_fwd_kernel.hpp

add_rmsnorm2d_rdquant_fwd_pipeline_default_policy.hpp

add_rmsnorm2d_rdquant_fwd_pipeline_one_pass.hpp

add_rmsnorm2d_rdquant_fwd_pipeline_problem.hpp

add_rmsnorm2d_rdquant_fwd_pipeline_three_pass.hpp

generic_2d_block_shape.hpp

tensor_layout.hpp