/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 // Copyright (c) Advanced Micro Devices, Inc., or its affiliates.

2 // SPDX-License-Identifier: MIT

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  QRKSVS_ASYNC_TRLOAD_V3,

16 };

17 

18 template <BlockFmhaPipelineEnum>

19 struct BlockFmhaPipelineEnumToStr;

20 

21 template <>

22 struct BlockFmhaPipelineEnumToStr<BlockFmhaPipelineEnum::QRKSVS>

23 {

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

25 };

26 template <>

27 struct BlockFmhaPipelineEnumToStr<BlockFmhaPipelineEnum::QRKSVS_ASYNC>

28 {

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

30 };

31 template <>

32 struct BlockFmhaPipelineEnumToStr<BlockFmhaPipelineEnum::QSKSVS>

33 {

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

35 };

36 

37 template <>

38 struct BlockFmhaPipelineEnumToStr<BlockFmhaPipelineEnum::QRKSVS_ASYNC_TRLOAD>

39 {

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

41 };

42 

43 } // 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::QRKSVS_ASYNC_TRLOAD_V3

@ QRKSVS_ASYNC_TRLOAD_V3

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:19