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

Remove un-necessary HSTU_CHECK() callings

Browse Source
This commit is contained in:
Qianfeng Zhang
2025-09-13 16:34:02 +00:00
parent 2427426640
commit db62a9f47e
1 changed files with 0 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
example/ck_tile/18_hstu_attention/example_hstu_attention.cpp
View File

@@ -280,14 +280,6 @@ bool run(const ck_tile::ArgParser& arg_parser)
{
max_uih_seqlen = max(max_uih_seqlen, seq_lengths[i]);
};
// only consider num_batch values even if more values are provided by the user
for(int i = 0; i < num_batch; i++)
{
HSTU_CHECK(
seq_lengths[i] >= min_full_attn_seqlen,
"min_full_attn_seqlen should not be bigger than the sequence length of any batch!");
};
}
else
{
@@ -295,9 +287,6 @@ bool run(const ck_tile::ArgParser& arg_parser)
"sequence lengths for batched mode shoud have single element!");
uih_seqlen = seq_lengths[0];
max_uih_seqlen = uih_seqlen;
HSTU_CHECK(max_uih_seqlen >= min_full_attn_seqlen,
"minf_full_attn_seqlen should not be bigger than the sequence length!");
};
// the user input of max_uih_seqlen can either be ignored or be bigger than all uih_seqlens