"torchvision/csrc/cuda/vision_cuda.h" did not exist on "dc3ac2909c2b9a5f6874ba1f1041ef49bd5482d6"
layernorm2d.hpp 703 Bytes
Newer Older
rocking's avatar
rocking committed
1
// SPDX-License-Identifier: MIT
Po Yen, Chen's avatar
Po Yen, Chen committed
2
// Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
rocking's avatar
rocking committed
3
4
5
6

#pragma once

#include "ck_tile/ops/layernorm2d/kernel/layernorm2d_fwd_kernel.hpp"
ltqin's avatar
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"
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
rocking committed
13
#include "ck_tile/ops/common/tensor_layout.hpp"