/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/fmha/pipeline/block_fmha_pipeline_enum.hpp Source File

• develop
• 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  QRKSVS_ASYNC_TRLOAD,

15 };

16 

17 template <BlockFmhaPipelineEnum>

18 struct BlockFmhaPipelineEnumToStr;

19 

20 template <>

21 struct BlockFmhaPipelineEnumToStr<BlockFmhaPipelineEnum::QRKSVS>

22 {

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

24 };

25 template <>

26 struct BlockFmhaPipelineEnumToStr<BlockFmhaPipelineEnum::QRKSVS_ASYNC>

27 {

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

29 };

30 template <>

31 struct BlockFmhaPipelineEnumToStr<BlockFmhaPipelineEnum::QSKSVS>

32 {

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

34 };

35 

36 template <>

37 struct BlockFmhaPipelineEnumToStr<BlockFmhaPipelineEnum::QRKSVS_ASYNC_TRLOAD>

38 {

39  static constexpr const char* name = "qr_async_trload";

40 };

41 

42 } // 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::BlockFmhaPipelineEnum::QRKSVS_ASYNC_TRLOAD

@ QRKSVS_ASYNC_TRLOAD

ck_tile::BlockFmhaPipelineEnumToStr

Definition: block_fmha_pipeline_enum.hpp:18