/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/core/tensor/static_distributed_tensor.hpp File Reference

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/core/tensor/static_distributed_tensor.hpp File Reference#

Composable Kernel: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/core/tensor/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

 ck_tile
 
 ck_tile::detail
 

Functions

template<typename DataType , typename StaticTileDistribution >
constexpr CK_TILE_HOST_DEVICE auto ck_tile::make_static_distributed_tensor (const StaticTileDistribution &)
 
template<typename DataType , typename StaticTileDistribution , typename ThreadBuffer >
constexpr CK_TILE_HOST_DEVICE auto ck_tile::make_static_distributed_tensor (const StaticTileDistribution &, ThreadBuffer &&thread_buffer_)
 
template<typename StaticTileDistribution , typename DistributedIndices >
constexpr CK_TILE_HOST_DEVICE 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>
constexpr CK_TILE_HOST_DEVICE 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