ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-06 15:54:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Rename RotaryEmbeddingEnum

Browse Source
This commit is contained in:
PoYen, Chen
2024-07-23 07:50:50 +00:00
parent d4606cf3c3
commit 2192bbc68a
8 changed files with 27 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
example/ck_tile/01_fmha/codegen/cpp_symbol_map.py
View File

@@ -67,9 +67,9 @@ BIAS_CHECK_MAP = {
}
ROPE_MAP = {
"no" : "ck_tile::BlockRotaryEmbeddingEnum::NONE",
"inter" : "ck_tile::BlockRotaryEmbeddingEnum::INTERLEAVED",
"half" : "ck_tile::BlockRotaryEmbeddingEnum::HALF_ROTATED"
"no" : "ck_tile::RotaryEmbeddingEnum::NONE",
"inter" : "ck_tile::RotaryEmbeddingEnum::INTERLEAVED",
"half" : "ck_tile::RotaryEmbeddingEnum::HALF_ROTATED"
}
# TODO: avoid duplication

2
example/ck_tile/01_fmha/fmha_fwd.hpp
View File

@@ -642,7 +642,7 @@ template <ck_tile::index_t HDim_,
bool kPadSk_,
bool kPadD_,
bool kPadDv_,
ck_tile::BlockRotaryEmbeddingEnum RotaryEnum_>
ck_tile::RotaryEmbeddingEnum RotaryEnum_>
struct fmha_fwd_appendkv_traits_
{
static constexpr ck_tile::index_t HDim = HDim_;

2
example/ck_tile/01_fmha/rotary.hpp
View File

@@ -14,7 +14,7 @@
#include <random>
#include <tuple>
// keep sync with BlockRotaryEmbeddingEnum
// keep sync with RotaryEmbeddingEnum
enum class rope_enum
{
none = 0,