/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck_tile/ops/fmha/pipeline/tile_fmha_shape.hpp Source File#
tile_fmha_shape.hpp
Go to the documentation of this file.
106 static constexpr index_t kK4 = BlockTile::at(number<6>{}); // tile size along gemm4(dS@K) unroll
110 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:20
constexpr CK_TILE_HOST_DEVICE index_t reduce_on_sequence(Seq, Reduce f, number< Init >)
Definition: sequence.hpp:973
typename conditional< predicate, X, Y >::type conditional_t
Definition: functional.hpp:115
Definition: tile_fmha_shape.hpp:77
remove_cvref_t< Gemm2BlockWarps_ > Gemm2BlockWarps
Definition: tile_fmha_shape.hpp:83
static constexpr index_t NumWarps
Definition: tile_fmha_shape.hpp:90
remove_cvref_t< BlockTile_ > BlockTile
Definition: tile_fmha_shape.hpp:78
remove_cvref_t< Gemm0WarpTile_ > Gemm0WarpTile
Definition: tile_fmha_shape.hpp:80
remove_cvref_t< Gemm2WarpTile_ > Gemm2WarpTile
Definition: tile_fmha_shape.hpp:84
remove_cvref_t< Gemm4WarpTile_ > Gemm4WarpTile
Definition: tile_fmha_shape.hpp:88
static constexpr index_t kQKHeaddim
Definition: tile_fmha_shape.hpp:107
remove_cvref_t< Gemm0BlockWarps_ > Gemm0BlockWarps
Definition: tile_fmha_shape.hpp:79
static constexpr index_t kVHeaddim
Definition: tile_fmha_shape.hpp:110
remove_cvref_t< Gemm3BlockWarps_ > Gemm3BlockWarps
Definition: tile_fmha_shape.hpp:85
remove_cvref_t< Gemm4BlockWarps_ > Gemm4BlockWarps
Definition: tile_fmha_shape.hpp:87
remove_cvref_t< Gemm3WarpTile_ > Gemm3WarpTile
Definition: tile_fmha_shape.hpp:86
remove_cvref_t< Gemm1WarpTile_ > Gemm1WarpTile
Definition: tile_fmha_shape.hpp:82
remove_cvref_t< Gemm1BlockWarps_ > Gemm1BlockWarps
Definition: tile_fmha_shape.hpp:81
Definition: tile_fmha_shape.hpp:31
static constexpr bool IsVLayoutRowMajor
Definition: tile_fmha_shape.hpp:59
std::conditional_t< IsVLayoutRowMajor, ck_tile::tensor_layout::gemm::RowMajor, ck_tile::tensor_layout::gemm::ColumnMajor > VLayout
Definition: tile_fmha_shape.hpp:62
static constexpr index_t kQKHeaddim
Definition: tile_fmha_shape.hpp:51
remove_cvref_t< Gemm1BlockWarps_ > Gemm1BlockWarps
Definition: tile_fmha_shape.hpp:35
static constexpr index_t NumGemm0Warps
Definition: tile_fmha_shape.hpp:38
remove_cvref_t< Gemm1WarpTile_ > Gemm1WarpTile
Definition: tile_fmha_shape.hpp:36
remove_cvref_t< Gemm0WarpTile_ > Gemm0WarpTile
Definition: tile_fmha_shape.hpp:34
static constexpr index_t kSubQKHeaddim
Definition: tile_fmha_shape.hpp:56
remove_cvref_t< Gemm0BlockWarps_ > Gemm0BlockWarps
Definition: tile_fmha_shape.hpp:33
remove_cvref_t< BlockTile_ > BlockTile
Definition: tile_fmha_shape.hpp:32
static constexpr index_t NumGemm1Warps
Definition: tile_fmha_shape.hpp:40
Definition: integral_constant.hpp:13
Definition: math.hpp:98
Definition: tensor_layout.hpp:22
Definition: tensor_layout.hpp:17