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

• develop
• include
• ck_tile
• ops
• fmha
• block

block_attention_quant_scale_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 #include <string>

7 

8 namespace ck_tile {

9 

10 // This class is used for codegen pattern matching

11 enum class BlockAttentionQuantScaleEnum

12 {

13  NO_SCALE = 0,

14  PERTENSOR = 1,

15 };

16 

17 template <BlockAttentionQuantScaleEnum>

18 struct BlockAttentionQuantScaleEnumToStr;

19 

20 template <>

21 struct BlockAttentionQuantScaleEnumToStr<BlockAttentionQuantScaleEnum::NO_SCALE>

22 {

23  static constexpr const char* name = "";

24 };

25 template <>

26 struct BlockAttentionQuantScaleEnumToStr<BlockAttentionQuantScaleEnum::PERTENSOR>

27 {

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

29 };

30 

31 } // namespace ck_tile

ck_tile

Definition: cluster_descriptor.hpp:13

ck_tile::BlockAttentionQuantScaleEnum

BlockAttentionQuantScaleEnum

Definition: block_attention_quant_scale_enum.hpp:12

ck_tile::BlockAttentionQuantScaleEnum::NO_SCALE

@ NO_SCALE

ck_tile::BlockAttentionQuantScaleEnum::PERTENSOR

@ PERTENSOR

ck_tile::BlockAttentionQuantScaleEnumToStr

Definition: block_attention_quant_scale_enum.hpp:18