load_tile_transpose.hpp File Reference

load_tile_transpose.hpp File Reference#

Composable Kernel: load_tile_transpose.hpp File Reference
load_tile_transpose.hpp File Reference

Go to the source code of this file.

Classes

struct  ck_tile::util::is_sequence_suffix< Suffix, Sequence >
struct  ck_tile::util::is_sequence_suffix< sequence<>, sequence< Xs... > >
struct  ck_tile::DefaultTranspose< DataType >
struct  ck_tile::DefaultTranspose< DataType >::Quad16< LaneGroupSize >
struct  ck_tile::DefaultTranspose< DataType >::Quad8< LaneGroupSize >
struct  ck_tile::DefaultTranspose< DataType >::ValidationTraitsImpl< InDstrEncode, ReverseDirection, LaneGroupSize >
struct  ck_tile::DefaultTranspose< DataType >::ValidationTraits< InDstrEncode, ReverseDirection >
struct  ck_tile::TransposeTileDistrChecker< TileDistribution_, DataType_, Policy >
struct  ck_tile::TransposeTileDistributionTraits< TileDistributionEncoding_, DataType_, Policy, ReverseDirection >

Namespaces

namespace  ck_tile
namespace  ck_tile::util

Typedefs

template<typename TileDistributionEncoding_, typename DataType_, typename Policy = DefaultTranspose<DataType_>>
using ck_tile::OutputTileDistributionTraits
template<typename TileDistributionEncoding_, typename DataType_, typename Policy = DefaultTranspose<DataType_>>
using ck_tile::InputTileDistributionTraits

Functions

constexpr int ck_tile::DS_READ_TR_SIZE ()
template<typename InnerEncode, index_t kLeadIterPerWarp, index_t kSecondIterPerWarp, index_t kLeadNumWarps, index_t kSecondNumWarps>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::InputTileDistributionEncoding ()
template<typename BottomTensorView_, typename WindowLengths_, typename TileDistribution_, index_t NumCoord, typename Policy = DefaultTranspose<typename BottomTensorView_::DataType>, typename = std::enable_if_t<TransposeTileDistrChecker<TileDistribution_, typename BottomTensorView_::DataType, Policy>::distr_encoding_valid, Policy>>
CK_TILE_DEVICE auto ck_tile::load_tile_transpose (const tile_window_with_static_distribution< BottomTensorView_, WindowLengths_, TileDistribution_, NumCoord > &tile_window)
 transpose loads tile from a tensor and returns the resulting tensor with a new (transposed) tile distribution. use SFINAE to ensure the tile distribution encoding is valid.

Variables

template<typename Suffix, typename Sequence>
constexpr bool ck_tile::util::is_sequence_suffix_v = is_sequence_suffix<Suffix, Sequence>::value