tile_distribution.hpp File Reference

tile_distribution.hpp File Reference#

Composable Kernel: tile_distribution.hpp File Reference
Classes | Namespaces | Functions
tile_distribution.hpp File Reference
#include "ck_tile/core/arch/arch.hpp"
#include "ck_tile/core/config.hpp"
#include "ck_tile/core/container/array.hpp"
#include "ck_tile/core/container/sequence.hpp"
#include "ck_tile/core/container/tuple.hpp"
#include "ck_tile/core/container/container_helper.hpp"
#include "ck_tile/core/container/meta_data_buffer.hpp"
#include "ck_tile/core/tensor/tensor_adaptor.hpp"
#include "ck_tile/core/tensor/tile_distribution_encoding.hpp"
#include "ck_tile/core/utility/functional.hpp"
#include "ck_tile/core/utility/type_traits.hpp"

Go to the source code of this file.

Classes

struct  ck_tile::tile_distributed_span< PartialHsLengths >
struct  ck_tile::tile_distributed_index< PartialHsIndices >
struct  ck_tile::tile_distribution< PsYs2XsAdaptor_, Ys2DDescriptor_, StaticTileDistributionEncoding_, TileDistributionDetail_ >
struct  ck_tile::detail::tile_distribution_detail< RhMajorMinor2AdaptorHiddenIdss >

Namespaces

namespace  ck_tile
namespace  ck_tile::detail

Functions

template<typename Distribution>
CK_TILE_HOST_DEVICE auto ck_tile::detail::get_partition_index (Distribution)
template<index_t... Is>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::detail::make_tile_distributed_span (sequence< Is... >)
template<index_t... Is>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::detail::make_tile_distributed_index (sequence< Is... >)
template<index_t NDimMax>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::detail::make_sequential_index (index_t ibegin, index_t iend)
template<typename StaticTileDistributionEncoding_>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::detail::make_adaptor_encoding_for_tile_distribution (StaticTileDistributionEncoding_)
template<typename StaticTileDistributionEncoding_>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::make_static_tile_distribution (StaticTileDistributionEncoding_)
template<typename Distribution, index_t... XSliceBegins, index_t... XSliceEnds>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::detail::slice_distribution_from_x (Distribution, sequence< XSliceBegins... > x_slice_begins, sequence< XSliceEnds... > x_slice_ends)
template<typename PsYs2XsAdaptor_, typename Ys2DDescriptor_, typename StaticTileDistributionEncoding_, typename TileDistributionDetail_>
CK_TILE_HOST_DEVICE void ck_tile::print (const tile_distribution< PsYs2XsAdaptor_, Ys2DDescriptor_, StaticTileDistributionEncoding_, TileDistributionDetail_ > &distribution)