From 9940bd07f6da1f90693bcd3cb27878228de22d32 Mon Sep 17 00:00:00 2001 From: Juuso Korhonen <40278371+juuso-oskari@users.noreply.github.com> Date: Thu, 16 Oct 2025 11:23:46 +0000 Subject: [PATCH] fix order in mask caller --- .../ops/unified_attention/kernel/unified_attention_kernel.hpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/include/ck_tile/ops/unified_attention/kernel/unified_attention_kernel.hpp b/include/ck_tile/ops/unified_attention/kernel/unified_attention_kernel.hpp index 57061dbe18..e19f8c4fce 100644 --- a/include/ck_tile/ops/unified_attention/kernel/unified_attention_kernel.hpp +++ b/include/ck_tile/ops/unified_attention/kernel/unified_attention_kernel.hpp @@ -460,10 +460,10 @@ struct UnifiedAttentionKernel FmhaMask mask = [&]() { if constexpr(kHasMask) return ck_tile::make_generic_attention_mask_from_lr_window( - cur_batch_query_len, // x (i.e. extend) seq_len - cur_batch_query_len, // y (i.e. context) - cur_batch_query_len, // x_total + cur_batch_query_len, // x (i.e. extend) seq_len, // y_total (x + y) + cur_batch_query_len, // x_total num_queries_per_kv, // the same sequence index is repeated num_queries_per_kv times along x dim of the tile kargs.mask_type == GenericAttentionMaskEnum::MASK_FROM_TOP_LEFT); else