/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/fmha/kernel/fmha_fwd_appendkv_tile_partitioner.hpp Source File#
fmha_fwd_appendkv_tile_partitioner.hpp
Go to the documentation of this file.
Definition: cluster_descriptor.hpp:13
constexpr CK_TILE_HOST_DEVICE auto integer_divide_ceil(X x, Y y)
Definition: math.hpp:149
constexpr CK_TILE_HOST_DEVICE auto make_tuple(Xs &&... xs)
Definition: tuple.hpp:360
Definition: fmha_fwd_appendkv_tile_partitioner.hpp:12
static constexpr ck_tile::index_t kN1
Definition: fmha_fwd_appendkv_tile_partitioner.hpp:16
CK_TILE_DEVICE auto operator()()
Definition: fmha_fwd_appendkv_tile_partitioner.hpp:32
static constexpr ck_tile::index_t kK0
Definition: fmha_fwd_appendkv_tile_partitioner.hpp:15
static constexpr CK_TILE_HOST auto GridSize(ck_tile::index_t batch_size, ck_tile::index_t nhead, ck_tile::index_t seqlen_q, ck_tile::index_t seqlen_knew)
Definition: fmha_fwd_appendkv_tile_partitioner.hpp:20
static constexpr ck_tile::index_t kM0
Definition: fmha_fwd_appendkv_tile_partitioner.hpp:13
static constexpr ck_tile::index_t kN0
Definition: fmha_fwd_appendkv_tile_partitioner.hpp:14