mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-07-14 02:57:45 +00:00
33 lines
1.1 KiB
C++
33 lines
1.1 KiB
C++
// SPDX-License-Identifier: MIT
|
|
// Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "ck/utility/common_header.hpp"
|
|
#include "ck/tensor_description/tensor_descriptor.hpp"
|
|
#include "ck/tensor_description/tensor_descriptor_helper.hpp"
|
|
#include "ck/tensor_description/tensor_adaptor.hpp"
|
|
#include "ck/tensor_description/tensor_space_filling_curve.hpp"
|
|
|
|
#include "ck/tile_program/tile/tile_distribution.hpp"
|
|
#include "ck/tile_program/tile/tile_window.hpp"
|
|
#include "ck/tile_program/tile/static_distributed_tensor.hpp"
|
|
|
|
namespace ck {
|
|
namespace tile_program {
|
|
|
|
template <typename BottomTensorView_,
|
|
typename WindowLengths_,
|
|
typename TileDistribution_,
|
|
index_t NumCoord>
|
|
__device__ auto load_tile(const TileWindowWithStaticDistribution<BottomTensorView_,
|
|
WindowLengths_,
|
|
TileDistribution_,
|
|
NumCoord>& tile_window)
|
|
{
|
|
return tile_window.Load();
|
|
}
|
|
|
|
} // namespace tile_program
|
|
} // namespace ck
|