ElementWiseShape< BlockWarps, BlockTile, WarpTile, ComputeDataType > Struct Template Reference

• ck_tile
• ElementWiseShape

Static Public Attributes | List of all members

ck_tile::ElementWiseShape< BlockWarps, BlockTile, WarpTile, ComputeDataType > Struct Template Reference

#include <elementwise_shape.hpp>

Static Public Attributes
static constexpr index_t	kBlockM = BlockTile::at(number<0>{})
static constexpr index_t	kWarpM = WarpTile::at(number<0>{})
static constexpr index_t	kVectorM
static constexpr index_t	kWarpPerBlockM = BlockWarps::at(number<0>{})
static constexpr index_t	kThreadPerWarpM = get_warp_size()
static constexpr index_t	kRepeatM = kBlockM / (kWarpPerBlockM * kVectorM * kThreadPerWarpM)
static constexpr index_t	kBlockSize

Member Data Documentation

◆ kBlockM

template<typename BlockWarps, typename BlockTile, typename WarpTile, typename ComputeDataType>

index_t ck_tile::ElementWiseShape< BlockWarps, BlockTile, WarpTile, ComputeDataType >::kBlockM = BlockTile::at(number<0>{})

staticconstexpr

◆ kBlockSize

template<typename BlockWarps, typename BlockTile, typename WarpTile, typename ComputeDataType>

index_t ck_tile::ElementWiseShape< BlockWarps, BlockTile, WarpTile, ComputeDataType >::kBlockSize

staticconstexpr

Initial value:

=

ck_tile::get_warp_size() * reduce_on_sequence(BlockWarps{}, multiplies{}, number<1>{})

ck_tile::get_warp_size

CK_TILE_HOST_DEVICE constexpr index_t get_warp_size()

Definition arch.hpp:63

ck_tile::number

constant< v > number

Definition tile/core/numeric/integral_constant.hpp:37

ck_tile::reduce_on_sequence

CK_TILE_HOST_DEVICE constexpr index_t reduce_on_sequence(Seq, Reduce f, number< Init >)

Definition tile/core/container/sequence.hpp:982

ck_tile::multiplies

Definition tile/core/numeric/math.hpp:98

◆ kRepeatM

template<typename BlockWarps, typename BlockTile, typename WarpTile, typename ComputeDataType>

index_t ck_tile::ElementWiseShape< BlockWarps, BlockTile, WarpTile, ComputeDataType >::kRepeatM = kBlockM / (kWarpPerBlockM * kVectorM * kThreadPerWarpM)

staticconstexpr

◆ kThreadPerWarpM

template<typename BlockWarps, typename BlockTile, typename WarpTile, typename ComputeDataType>

index_t ck_tile::ElementWiseShape< BlockWarps, BlockTile, WarpTile, ComputeDataType >::kThreadPerWarpM = get_warp_size()

staticconstexpr

◆ kVectorM

template<typename BlockWarps, typename BlockTile, typename WarpTile, typename ComputeDataType>

index_t ck_tile::ElementWiseShape< BlockWarps, BlockTile, WarpTile, ComputeDataType >::kVectorM

staticconstexpr

Initial value:

=

min(static_cast<index_t>(16 / sizeof(ComputeDataType)), kWarpM / get_warp_size())

ck_tile::min

CK_TILE_HOST_DEVICE constexpr T min(T x)

Definition tile/core/numeric/math.hpp:210

ck_tile::index_t

int32_t index_t

Definition integer.hpp:9

ck_tile::ElementWiseShape::kWarpM

static constexpr index_t kWarpM

Definition elementwise_shape.hpp:15

◆ kWarpM

template<typename BlockWarps, typename BlockTile, typename WarpTile, typename ComputeDataType>

index_t ck_tile::ElementWiseShape< BlockWarps, BlockTile, WarpTile, ComputeDataType >::kWarpM = WarpTile::at(number<0>{})

staticconstexpr

◆ kWarpPerBlockM

template<typename BlockWarps, typename BlockTile, typename WarpTile, typename ComputeDataType>

index_t ck_tile::ElementWiseShape< BlockWarps, BlockTile, WarpTile, ComputeDataType >::kWarpPerBlockM = BlockWarps::at(number<0>{})

staticconstexpr

---

The documentation for this struct was generated from the following file:

• elementwise_shape.hpp