/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/fmha/pipeline/tile_fmha_shape.hpp Source File#
tile_fmha_shape.hpp
Go to the documentation of this file.
109 static constexpr index_t kK4 = BlockTile::at(number<6>{}); // tile size along gemm4(dS@K) unroll
113 static constexpr index_t kVHeaddim = BlockTile::at(number<8>{}); // V headdim, used for pipeline
Definition: cluster_descriptor.hpp:13
__host__ __device__ multiplies() -> multiplies< void, void >
FIXME: create macro to replace 'host device' and nothing more.
remove_cv_t< std::remove_reference_t< T > > remove_cvref_t
Definition: type_traits.hpp:21
constexpr CK_TILE_HOST_DEVICE index_t reduce_on_sequence(Seq, Reduce f, number< Init >)
Definition: sequence.hpp:979
typename conditional< predicate, X, Y >::type conditional_t
Definition: functional.hpp:115
Definition: tile_fmha_shape.hpp:80
static constexpr index_t kQKHeaddim
Definition: tile_fmha_shape.hpp:110
remove_cvref_t< Gemm0BlockWarps_ > Gemm0BlockWarps
Definition: tile_fmha_shape.hpp:82
remove_cvref_t< Gemm1WarpTile_ > Gemm1WarpTile
Definition: tile_fmha_shape.hpp:85
static constexpr index_t kMaxSeqLenQ
Definition: tile_fmha_shape.hpp:116
remove_cvref_t< Gemm4WarpTile_ > Gemm4WarpTile
Definition: tile_fmha_shape.hpp:91
remove_cvref_t< Gemm4BlockWarps_ > Gemm4BlockWarps
Definition: tile_fmha_shape.hpp:90
static constexpr index_t kVHeaddim
Definition: tile_fmha_shape.hpp:113
remove_cvref_t< Gemm2BlockWarps_ > Gemm2BlockWarps
Definition: tile_fmha_shape.hpp:86
remove_cvref_t< Gemm3BlockWarps_ > Gemm3BlockWarps
Definition: tile_fmha_shape.hpp:88
remove_cvref_t< Gemm0WarpTile_ > Gemm0WarpTile
Definition: tile_fmha_shape.hpp:83
remove_cvref_t< Gemm2WarpTile_ > Gemm2WarpTile
Definition: tile_fmha_shape.hpp:87
remove_cvref_t< BlockTile_ > BlockTile
Definition: tile_fmha_shape.hpp:81
static constexpr index_t NumWarps
Definition: tile_fmha_shape.hpp:93
remove_cvref_t< Gemm3WarpTile_ > Gemm3WarpTile
Definition: tile_fmha_shape.hpp:89
remove_cvref_t< Gemm1BlockWarps_ > Gemm1BlockWarps
Definition: tile_fmha_shape.hpp:84
Definition: tile_fmha_shape.hpp:33
static constexpr bool IsVLayoutRowMajor
Definition: tile_fmha_shape.hpp:61
std::conditional_t< IsVLayoutRowMajor, ck_tile::tensor_layout::gemm::RowMajor, ck_tile::tensor_layout::gemm::ColumnMajor > VLayout
Definition: tile_fmha_shape.hpp:64
static constexpr index_t kQKHeaddim
Definition: tile_fmha_shape.hpp:53
remove_cvref_t< Gemm1BlockWarps_ > Gemm1BlockWarps
Definition: tile_fmha_shape.hpp:37
static constexpr index_t NumGemm0Warps
Definition: tile_fmha_shape.hpp:40
remove_cvref_t< Gemm1WarpTile_ > Gemm1WarpTile
Definition: tile_fmha_shape.hpp:38
remove_cvref_t< Gemm0WarpTile_ > Gemm0WarpTile
Definition: tile_fmha_shape.hpp:36
static constexpr index_t kSubQKHeaddim
Definition: tile_fmha_shape.hpp:58
remove_cvref_t< Gemm0BlockWarps_ > Gemm0BlockWarps
Definition: tile_fmha_shape.hpp:35
remove_cvref_t< BlockTile_ > BlockTile
Definition: tile_fmha_shape.hpp:34
static constexpr index_t NumGemm1Warps
Definition: tile_fmha_shape.hpp:42
Definition: integral_constant.hpp:13
Definition: math.hpp:98
Definition: tensor_layout.hpp:22
Definition: tensor_layout.hpp:17