Fix sequence dim length for o_dram descriptor in the kernel

2026-05-17 19:40:04 +00:00 · 2025-05-10 16:02:52 +00:00
parent 1d1dd8f1eb
commit 79cd1f0653
1 changed files with 1 additions and 1 deletions
--- a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_kernel.hpp
+++ b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_kernel.hpp
@@ -749,7 +749,7 @@ struct HstuAttentionFwdKernel
        auto o_dram = [&]() {
            const auto o_dram_naive = make_naive_tensor_view<address_space_enum::global>(
                o_ptr,
-                make_tuple(mask.max_uih_len, kargs.hdim_v),
+                make_tuple(kargs.seqlen, kargs.hdim_v),
                make_tuple(kargs.seq_stride_o, 1),
                number<HstuAttentionPipeline::kAlignmentO>{},
                number<1>{});