add_rmsnorm2d_rdquant.hpp Source File

add_rmsnorm2d_rdquant.hpp Source File#

Composable Kernel: add_rmsnorm2d_rdquant.hpp Source File
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/load_interleaved_pk_type.hpp"
13#include "ck_tile/ops/common/streamk_common.hpp"
14#include "ck_tile/ops/common/tensor_layout.hpp"
15#include "ck_tile/ops/common/utils.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
load_interleaved_pk_type.hpp
streamk_common.hpp
tensor_layout.hpp