include/ck_tile/ops/fmha/pipeline/block_fmha_pipeline_enum.hpp Source File

• include
• ck_tile
• ops
• fmha
• pipeline

block_fmha_pipeline_enum.hpp

Go to the documentation of this file.

1 // SPDX-License-Identifier: MIT

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

3 

4 #pragma once

5 

6 namespace ck_tile {

7 

8 // This class is used for codegen pattern matching

9 enum class BlockFmhaPipelineEnum

10 {

11  QRKSVS = 0,

12  QRKSVS_ASYNC,

13  QSKSVS,

14 };

15 

16 template <BlockFmhaPipelineEnum>

17 struct BlockFmhaPipelineEnumToStr;

18 

19 template <>

20 struct BlockFmhaPipelineEnumToStr<BlockFmhaPipelineEnum::QRKSVS>

21 {

22  static constexpr const char* name = "qr";

23 };

24 template <>

25 struct BlockFmhaPipelineEnumToStr<BlockFmhaPipelineEnum::QRKSVS_ASYNC>

26 {

27  static constexpr const char* name = "qr_async";

28 };

29 template <>

30 struct BlockFmhaPipelineEnumToStr<BlockFmhaPipelineEnum::QSKSVS>

31 {

32  static constexpr const char* name = "qs";

33 };

34 

35 } // namespace ck_tile

ck_tile

Definition: cluster_descriptor.hpp:13

ck_tile::BlockFmhaPipelineEnum

BlockFmhaPipelineEnum

Definition: block_fmha_pipeline_enum.hpp:10

ck_tile::BlockFmhaPipelineEnum::QRKSVS

@ QRKSVS

ck_tile::BlockFmhaPipelineEnum::QSKSVS

@ QSKSVS

ck_tile::BlockFmhaPipelineEnum::QRKSVS_ASYNC

@ QRKSVS_ASYNC

ck_tile::BlockFmhaPipelineEnumToStr

Definition: block_fmha_pipeline_enum.hpp:17