From e8cd975d6a88126d0c14c96b1291a2cc370c3350 Mon Sep 17 00:00:00 2001 From: "PoYen, Chen" Date: Sun, 18 Aug 2024 19:00:05 +0000 Subject: [PATCH] Fix compilation errors --- example/ck_tile/01_fmha/fmha_fwd.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/example/ck_tile/01_fmha/fmha_fwd.cpp b/example/ck_tile/01_fmha/fmha_fwd.cpp index 2c48c79aeb..7a2559e633 100644 --- a/example/ck_tile/01_fmha/fmha_fwd.cpp +++ b/example/ck_tile/01_fmha/fmha_fwd.cpp @@ -730,8 +730,8 @@ bool run(const ck_tile::ArgParser& arg_parser) ck_tile::DeviceMem seqstart_k(seqstart_k_host.size() * sizeof(int32_t)); ck_tile::DeviceMem seqlen_k_buf( use_kvcache || 0 <= seqlen_kpads[0] ? seqlen_ks.size() * sizeof(int32_t) : 0); - ck_tile::DeviceMem cache_seqlen_k_buf(need_appendkv ? cache_seqlen_ks.size() * sizeof(int32_t) - : 0); + ck_tile::DeviceMem cache_seqlen_k_buf( + need_append_kvcache ? cache_seqlen_ks.size() * sizeof(int32_t) : 0); ck_tile::DeviceMem rotary_cos_buf(rotary_cos_host.get_element_space_size_in_bytes()); ck_tile::DeviceMem rotary_sin_buf(rotary_sin_host.get_element_space_size_in_bytes()); ck_tile::DeviceMem randval_buf(randval_host.get_element_space_size_in_bytes());