From 203e22bf9cc968da488a8d1cf482c854c90c459e Mon Sep 17 00:00:00 2001 From: Qianfeng Zhang Date: Tue, 22 Jul 2025 13:13:45 +0000 Subject: [PATCH] Change the seqlen_q dim padding setting for o_dram and bias_dram --- .../18_hstu_attention/hstu_attention_fwd_kernel.hpp | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_kernel.hpp b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_kernel.hpp index a3dae30ea4..b294a4a3e0 100644 --- a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_kernel.hpp +++ b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_kernel.hpp @@ -778,9 +778,8 @@ struct HstuAttentionFwdKernel number{}, number<1>{}); - return pad_tensor_view(bias_dram_naive, - bias_dram_window_lengths, - sequence{}); + return pad_tensor_view( + bias_dram_naive, bias_dram_window_lengths, sequence{}); }(); return make_tile_window(bias_dram, bias_dram_window_lengths, {i_m0, 0}); @@ -833,7 +832,7 @@ struct HstuAttentionFwdKernel return pad_tensor_view(o_dram_naive, make_tuple(number{}, number{}), - sequence{}); + sequence{}); }(); auto o_dram_window = make_tile_window(