ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-15 10:37:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change the seqlen_q dim padding setting for o_dram and bias_dram

Browse Source
This commit is contained in:
Qianfeng Zhang
2025-07-22 13:13:45 +00:00
parent ecf6a86d88
commit 203e22bf9c
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
example/ck_tile/18_hstu_attention/hstu_attention_fwd_kernel.hpp
View File

@@ -778,9 +778,8 @@ struct HstuAttentionFwdKernel
number<HstuAttentionPipeline::kAlignmentBias>{},
number<1>{});
return pad_tensor_view(bias_dram_naive,
bias_dram_window_lengths,
sequence<kPadSeqLenQ, kPadSeqLenK>{});
return pad_tensor_view(
bias_dram_naive, bias_dram_window_lengths, sequence<false, kPadSeqLenK>{});
}();
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<HstuAttentionPipeline::kM0>{},
number<HstuAttentionPipeline::kN1>{}),
sequence<kPadSeqLenQ, kPadHeadDimV>{});
sequence<false, kPadHeadDimV>{});
}();
auto o_dram_window = make_tile_window(