static_distributed_tensor.hpp File Reference

static_distributed_tensor.hpp File Reference#

Composable Kernel: static_distributed_tensor.hpp File Reference
static_distributed_tensor.hpp File Reference

Go to the source code of this file.

Classes

struct  ck_tile::static_distributed_tensor< DataType_, StaticTileDistribution_ >
struct  ck_tile::detail::is_similiar_distributed_tensor< X, Y >
struct  ck_tile::detail::is_similiar_distributed_tensor< static_distributed_tensor< TypeX, DistX >, static_distributed_tensor< TypeY, DistY > >

Namespaces

namespace  ck_tile
namespace  ck_tile::detail

Functions

template<typename DataType, typename StaticTileDistribution>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::make_static_distributed_tensor (const StaticTileDistribution &)
template<typename DataType, typename StaticTileDistribution, typename ThreadBuffer>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::make_static_distributed_tensor (const StaticTileDistribution &, ThreadBuffer &&thread_buffer_)
template<typename StaticTileDistribution, typename DistributedIndices>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::get_x_indices_from_distributed_indices (StaticTileDistribution tile_distribution, DistributedIndices distributed_indices)
template<typename DataType, typename StaticTileDistribution, typename XIndicesPredicate>
CK_TILE_HOST_DEVICE void ck_tile::set_tile_if (static_distributed_tensor< DataType, StaticTileDistribution > &out_tensor, DataType value, XIndicesPredicate predicate)
template<typename YLengths, index_t XUnpacks>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::get_y_unpacks_from_x_unpacks (YLengths, number< XUnpacks >)

Variables

template<typename X, typename Y>
constexpr bool ck_tile::detail::is_similiar_distributed_tensor_v