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

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

Composable Kernel: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/3643/include/ck_tile/ops/fmha/block/block_attention_quant_scale_enum.hpp Source File
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  BLOCKSCALE,
16 };
17 
18 template <BlockAttentionQuantScaleEnum>
19 struct BlockAttentionQuantScaleEnumToStr;
20 
21 template <>
22 struct BlockAttentionQuantScaleEnumToStr<BlockAttentionQuantScaleEnum::NO_SCALE>
23 {
24  static constexpr const char* name = "";
25 };
26 template <>
27 struct BlockAttentionQuantScaleEnumToStr<BlockAttentionQuantScaleEnum::PERTENSOR>
28 {
29  static constexpr const char* name = "pertensor";
30 };
31 template <>
32 struct BlockAttentionQuantScaleEnumToStr<BlockAttentionQuantScaleEnum::BLOCKSCALE>
33 {
34  static constexpr const char* name = "blockscale";
35 };
36 
37 } // 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::BLOCKSCALE
@ BLOCKSCALE
ck_tile::BlockAttentionQuantScaleEnum::NO_SCALE
@ NO_SCALE
ck_tile::BlockAttentionQuantScaleEnum::PERTENSOR
@ PERTENSOR
ck_tile::BlockAttentionQuantScaleEnumToStr
Definition: block_attention_quant_scale_enum.hpp:19