BatchedTransposeProblem< DataType_, BlockTile, WarpLayout, kPadM_, kPadN_ > Struct Template Reference

BatchedTransposeProblem&lt; DataType_, BlockTile, WarpLayout, kPadM_, kPadN_ &gt; Struct Template Reference#

Composable Kernel: ck_tile::BatchedTransposeProblem< DataType_, BlockTile, WarpLayout, kPadM_, kPadN_ > Struct Template Reference
ck_tile::BatchedTransposeProblem< DataType_, BlockTile, WarpLayout, kPadM_, kPadN_ > Struct Template Reference

#include <batched_transpose_problem.hpp>

Public Types

using DataType = remove_cvref_t< DataType_ >
 

Static Public Attributes

static constexpr index_t kMPerWarp = WarpLayout::at(number<0>{})
 
static constexpr index_t kNPerWarp = WarpLayout::at(number<1>{})
 
static constexpr index_t kMPerBlock = BlockTile::at(number<0>{})
 
static constexpr index_t kNPerBlock = BlockTile::at(number<1>{})
 
static constexpr index_t kBlockSize = kMPerWarp * kNPerWarp * get_warp_size()
 
static constexpr bool kPadM = kPadM_
 
static constexpr bool kPadN = kPadN_
 
static constexpr index_t MaxLoadStoreSize = 16
 
static constexpr index_t VectorSizeInput = kPadN ? 1 : MaxLoadStoreSize / sizeof(DataType)
 
static constexpr index_t VectorSizeOutput = kPadM ? 1 : MaxLoadStoreSize / sizeof(DataType)
 

Member Typedef Documentation

◆ DataType

template<typename DataType_ , typename BlockTile , typename WarpLayout , bool kPadM_ = false, bool kPadN_ = false>
using ck_tile::BatchedTransposeProblem< DataType_, BlockTile, WarpLayout, kPadM_, kPadN_ >::DataType = remove_cvref_t<DataType_>

Member Data Documentation

◆ kBlockSize

template<typename DataType_ , typename BlockTile , typename WarpLayout , bool kPadM_ = false, bool kPadN_ = false>
constexpr index_t ck_tile::BatchedTransposeProblem< DataType_, BlockTile, WarpLayout, kPadM_, kPadN_ >::kBlockSize = kMPerWarp * kNPerWarp * get_warp_size()
staticconstexpr

◆ kMPerBlock

template<typename DataType_ , typename BlockTile , typename WarpLayout , bool kPadM_ = false, bool kPadN_ = false>
constexpr index_t ck_tile::BatchedTransposeProblem< DataType_, BlockTile, WarpLayout, kPadM_, kPadN_ >::kMPerBlock = BlockTile::at(number<0>{})
staticconstexpr

◆ kMPerWarp

template<typename DataType_ , typename BlockTile , typename WarpLayout , bool kPadM_ = false, bool kPadN_ = false>
constexpr index_t ck_tile::BatchedTransposeProblem< DataType_, BlockTile, WarpLayout, kPadM_, kPadN_ >::kMPerWarp = WarpLayout::at(number<0>{})
staticconstexpr

◆ kNPerBlock

template<typename DataType_ , typename BlockTile , typename WarpLayout , bool kPadM_ = false, bool kPadN_ = false>
constexpr index_t ck_tile::BatchedTransposeProblem< DataType_, BlockTile, WarpLayout, kPadM_, kPadN_ >::kNPerBlock = BlockTile::at(number<1>{})
staticconstexpr

◆ kNPerWarp

template<typename DataType_ , typename BlockTile , typename WarpLayout , bool kPadM_ = false, bool kPadN_ = false>
constexpr index_t ck_tile::BatchedTransposeProblem< DataType_, BlockTile, WarpLayout, kPadM_, kPadN_ >::kNPerWarp = WarpLayout::at(number<1>{})
staticconstexpr

◆ kPadM

template<typename DataType_ , typename BlockTile , typename WarpLayout , bool kPadM_ = false, bool kPadN_ = false>
constexpr bool ck_tile::BatchedTransposeProblem< DataType_, BlockTile, WarpLayout, kPadM_, kPadN_ >::kPadM = kPadM_
staticconstexpr

◆ kPadN

template<typename DataType_ , typename BlockTile , typename WarpLayout , bool kPadM_ = false, bool kPadN_ = false>
constexpr bool ck_tile::BatchedTransposeProblem< DataType_, BlockTile, WarpLayout, kPadM_, kPadN_ >::kPadN = kPadN_
staticconstexpr

◆ MaxLoadStoreSize

template<typename DataType_ , typename BlockTile , typename WarpLayout , bool kPadM_ = false, bool kPadN_ = false>
constexpr index_t ck_tile::BatchedTransposeProblem< DataType_, BlockTile, WarpLayout, kPadM_, kPadN_ >::MaxLoadStoreSize = 16
staticconstexpr

◆ VectorSizeInput

template<typename DataType_ , typename BlockTile , typename WarpLayout , bool kPadM_ = false, bool kPadN_ = false>
constexpr index_t ck_tile::BatchedTransposeProblem< DataType_, BlockTile, WarpLayout, kPadM_, kPadN_ >::VectorSizeInput = kPadN ? 1 : MaxLoadStoreSize / sizeof(DataType)
staticconstexpr

◆ VectorSizeOutput

template<typename DataType_ , typename BlockTile , typename WarpLayout , bool kPadM_ = false, bool kPadN_ = false>
constexpr index_t ck_tile::BatchedTransposeProblem< DataType_, BlockTile, WarpLayout, kPadM_, kPadN_ >::VectorSizeOutput = kPadM ? 1 : MaxLoadStoreSize / sizeof(DataType)
staticconstexpr

The documentation for this struct was generated from the following file:
  • /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/batched_transpose/pipeline/batched_transpose_problem.hpp