Skip to content

GitLab

"sgl-kernel/tests/test_trt_allreduce.py" did not exist on "6cb3974e77524ee2b291919ca6a8b55547bca8e0"
Switch branch/tag
Find file Normal viewHistoryPermalink
layernorm2d.hpp 703 Bytes
Newer Older
rocking's avatar
layernorm2d forward (#1339)  
rocking committed
1
2
3
4
5
6 
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved.

#pragma once

#include "ck_tile/ops/layernorm2d/kernel/layernorm2d_fwd_kernel.hpp"
ltqin's avatar
update layernorm (#1570)  
ltqin committed
7
8
9
10 
#include "ck_tile/ops/layernorm2d/pipeline/layernorm2d_fwd_pipeline_default_policy.hpp"
#include "ck_tile/ops/layernorm2d/pipeline/layernorm2d_fwd_pipeline_one_pass.hpp"
#include "ck_tile/ops/layernorm2d/pipeline/layernorm2d_fwd_pipeline_problem.hpp"
#include "ck_tile/ops/layernorm2d/pipeline/layernorm2d_fwd_pipeline_two_pass.hpp"
carlushuang's avatar
[CK_TILE] layernorm support fused-quant/fused-add (#1604)  
carlushuang committed
11
12 
#include "ck_tile/ops/layernorm2d/pipeline/layernorm2d_fwd_traits.hpp"
#include "ck_tile/ops/common/generic_2d_block_shape.hpp"
rocking's avatar
layernorm2d forward (#1339)  
rocking committed
13 
#include "ck_tile/ops/common/tensor_layout.hpp"