From e23e6b57ecbdca647bc070b5b14198e8cf39a102 Mon Sep 17 00:00:00 2001 From: "PoYen, Chen" Date: Tue, 20 Aug 2024 05:59:46 +0000 Subject: [PATCH] Fix compilation errors --- .../ck_tile/ops/fmha/kernel/fmha_fwd_splitkv_kernel.hpp | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/include/ck_tile/ops/fmha/kernel/fmha_fwd_splitkv_kernel.hpp b/include/ck_tile/ops/fmha/kernel/fmha_fwd_splitkv_kernel.hpp index 04c85892ac..08a8d5bcf4 100644 --- a/include/ck_tile/ops/fmha/kernel/fmha_fwd_splitkv_kernel.hpp +++ b/include/ck_tile/ops/fmha/kernel/fmha_fwd_splitkv_kernel.hpp @@ -219,7 +219,6 @@ struct FmhaFwdSplitKVKernel ck_tile::index_t batch_stride_k; ck_tile::index_t batch_stride_v; - ck_tile::index_t batch_stride_lse_acc; }; using Kargs = std::conditional_t; @@ -297,7 +296,8 @@ struct FmhaFwdSplitKVKernel nhead_stride_v, nhead_stride_lse_acc, nhead_stride_o_acc, - batch_stride_lse_acc batch_stride_o_acc, + batch_stride_lse_acc, + batch_stride_o_acc, split_stride_lse_acc, split_stride_o_acc}, // args for common karg {}, // placeholder for bias @@ -377,7 +377,8 @@ struct FmhaFwdSplitKVKernel ck_tile::index_t nhead_stride_o_acc, ck_tile::index_t batch_stride_k, ck_tile::index_t batch_stride_v, - ck_tile::index_t batch_stride_lse_acc ck_tile::index_t batch_stride_o_acc, + ck_tile::index_t batch_stride_lse_acc, + ck_tile::index_t batch_stride_o_acc, ck_tile::index_t split_stride_lse_acc, ck_tile::index_t split_stride_o_acc, ck_tile::index_t window_size_left,