"solve_ca_node1.sh" did not exist on "8c9dc81d5e687fa6268d6f2d06e5a33353469ed1"
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"
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"