block_fmha_pipeline_qr_ks_vs_default_policy.hpp Source File

• include
• ck_tile
• ops
• fmha
• pipeline

block_fmha_pipeline_qr_ks_vs_default_policy.hpp

Go to the documentation of this file.

1// SPDX-License-Identifier: MIT

2// Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved.

3

4#pragma once

5

6#include "ck_tile/core.hpp"

7#include "ck_tile/ops/fmha/pipeline/block_fmha_pipeline_qx_ks_vs_custom_policy.hpp"

8

9namespace ck_tile {

10

11using BlockFmhaPipelineQRKSVSDefaultPolicy =

12 BlockFmhaPipelineQXKSVSCustomPolicy</* QLoadOnce = */ true,

13 /* AsyncCopy = */ false,

14 /* NumPrefetchK = */ 1,

15 /* NumPrefetchV = */ 1>;

16

17} // namespace ck_tile

block_fmha_pipeline_qx_ks_vs_custom_policy.hpp

core.hpp

ck_tile

Definition tile/core/algorithm/cluster_descriptor.hpp:13

ck_tile::BlockFmhaPipelineQRKSVSDefaultPolicy

BlockFmhaPipelineQXKSVSCustomPolicy< true, false, 1, 1 > BlockFmhaPipelineQRKSVSDefaultPolicy

Definition block_fmha_pipeline_qr_ks_vs_default_policy.hpp:11

ck_tile::BlockFmhaPipelineQXKSVSCustomPolicy

Definition block_fmha_pipeline_qx_ks_vs_custom_policy.hpp:266