welford.hpp 405 Bytes
Newer Older
rocking's avatar
rocking committed
1
2
3
4
5
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved.

#pragma once

ltqin's avatar
ltqin committed
6
7
#include "ck_tile/ops/welford/block/block_welford.hpp"
#include "ck_tile/ops/welford/block/block_welford_problem.hpp"
rocking's avatar
rocking committed
8
#include "ck_tile/ops/welford/thread/thread_welford.hpp"
carlushuang's avatar
carlushuang committed
9
#include "ck_tile/ops/common/generic_2d_block_shape.hpp"
rocking's avatar
rocking committed
10
#include "ck_tile/ops/common/tensor_layout.hpp"